Spaces:
Running
Running
Update app.py
Browse files
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=
|
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 |
|