Spaces:

Ron0420
/

EfficientNetV2_Deepfakes_Image_Detector

Runtime error

Ron0420 commited on Jan 10, 2022

Commit

58d9bad

1 Parent(s): 50f42a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,9 +25,9 @@ zip_ref = zipfile.ZipFile(local_zip, 'r')
 zip_ref.extractall('deepfakes-test-images')
 zip_ref.close()
-model_b0 = tf.keras.models.load_model("FINAL-EFFICIENTNETV2-B0")
-model_s = tf.keras.models.load_model("FINAL-EFFICIENTNETV2-S")
 detector = MTCNN()

 zip_ref.extractall('deepfakes-test-images')
 zip_ref.close()
+with tf.device('/cpu:0'):
+    model_b0 = tf.keras.models.load_model("FINAL-EFFICIENTNETV2-B0")
+    model_s = tf.keras.models.load_model("FINAL-EFFICIENTNETV2-S")
 detector = MTCNN()