Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,9 @@ detector = MTCNN()
|
|
34 |
|
35 |
def deepfakespredict(select_model, input_img ):
|
36 |
|
37 |
-
|
|
|
|
|
38 |
|
39 |
if select_model == "EfficientNetV2-B0":
|
40 |
model = model_b0
|
@@ -72,7 +74,7 @@ def deepfakespredict(select_model, input_img ):
|
|
72 |
|
73 |
title="EfficientNetV2 Deepfakes Image Detector"
|
74 |
description="This is a demo implementation of EfficientNetV2 Deepfakes Image Detector. To use it, simply upload your image, or click one of the examples to load them."
|
75 |
-
examples = [
|
76 |
['deepfakes-test-images/Fake-1.jpg'],
|
77 |
['deepfakes-test-images/Fake-2.jpg'],
|
78 |
['deepfakes-test-images/Fake-3.jpg'],
|
@@ -84,6 +86,7 @@ examples = [
|
|
84 |
['deepfakes-test-images/Real-3.jpg'],
|
85 |
['deepfakes-test-images/Real-4.jpg'],
|
86 |
['deepfakes-test-images/Real-5.jpg']
|
|
|
87 |
]
|
88 |
|
89 |
gr.Interface(deepfakespredict,
|
@@ -91,4 +94,5 @@ gr.Interface(deepfakespredict,
|
|
91 |
outputs=["text", gr.outputs.Image(type="pil", label="Detected face"), gr.outputs.Label(num_top_classes=None, type="auto", label="Confidence")],
|
92 |
title=title,
|
93 |
description=description
|
|
|
94 |
).launch()
|
|
|
34 |
|
35 |
def deepfakespredict(select_model, input_img ):
|
36 |
|
37 |
+
model = []
|
38 |
+
labels = ['real', 'fake']
|
39 |
+
pred = [0, 0]
|
40 |
|
41 |
if select_model == "EfficientNetV2-B0":
|
42 |
model = model_b0
|
|
|
74 |
|
75 |
title="EfficientNetV2 Deepfakes Image Detector"
|
76 |
description="This is a demo implementation of EfficientNetV2 Deepfakes Image Detector. To use it, simply upload your image, or click one of the examples to load them."
|
77 |
+
examples = [ [],[
|
78 |
['deepfakes-test-images/Fake-1.jpg'],
|
79 |
['deepfakes-test-images/Fake-2.jpg'],
|
80 |
['deepfakes-test-images/Fake-3.jpg'],
|
|
|
86 |
['deepfakes-test-images/Real-3.jpg'],
|
87 |
['deepfakes-test-images/Real-4.jpg'],
|
88 |
['deepfakes-test-images/Real-5.jpg']
|
89 |
+
]
|
90 |
]
|
91 |
|
92 |
gr.Interface(deepfakespredict,
|
|
|
94 |
outputs=["text", gr.outputs.Image(type="pil", label="Detected face"), gr.outputs.Label(num_top_classes=None, type="auto", label="Confidence")],
|
95 |
title=title,
|
96 |
description=description
|
97 |
+
examples=examples
|
98 |
).launch()
|