Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,9 +30,9 @@ def deepfakespredict(input_img, select_model):
|
|
| 30 |
|
| 31 |
tf.keras.backend.clear_session()
|
| 32 |
|
| 33 |
-
if select_model
|
| 34 |
model = model_b0
|
| 35 |
-
elif select_model
|
| 36 |
model = model_s
|
| 37 |
|
| 38 |
text =""
|
|
|
|
| 30 |
|
| 31 |
tf.keras.backend.clear_session()
|
| 32 |
|
| 33 |
+
if select_model == "EfficientNetV2-B0":
|
| 34 |
model = model_b0
|
| 35 |
+
elif select_model == "EfficientNetV2-B0":
|
| 36 |
model = model_s
|
| 37 |
|
| 38 |
text =""
|