Update app.py

app.py

@@ -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")
+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()