Hemg commited on
Commit
1c903c0
·
verified ·
1 Parent(s): 3fb5c0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import pipeline
2
  import gradio as gr
3
 
4
 
5
- modelName = "Deepfake-Detection"
6
  hfUser = "Hemg"
7
 
8
 
@@ -31,7 +31,7 @@ def create_demo():
31
  demo = gr.Interface(
32
  fn=prediction_function,
33
  inputs=gr.Image(type="pil"),
34
- outputs=gr.Label(num_top_classes=3),
35
  )
36
  demo.launch()
37
 
 
2
  import gradio as gr
3
 
4
 
5
+ modelName = "Deepfake-image-Detection"
6
  hfUser = "Hemg"
7
 
8
 
 
31
  demo = gr.Interface(
32
  fn=prediction_function,
33
  inputs=gr.Image(type="pil"),
34
+ outputs=gr.Label(num_top_classes=2),
35
  )
36
  demo.launch()
37