Zeyadd-Mostaffa commited on
Commit
3724ecf
·
verified ·
1 Parent(s): 51e2b89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -7,8 +7,11 @@ from tensorflow.keras.applications.xception import preprocess_input as xcp_pre
7
  from tensorflow.keras.applications.efficientnet import preprocess_input as eff_pre
8
 
9
  # Load models
10
- xcp_model = load_model("xception_model.h5")
11
- eff_model = load_model("efficientnet_model.h5")
 
 
 
12
 
13
  # Load face detector
14
  detector = MTCNN()
 
7
  from tensorflow.keras.applications.efficientnet import preprocess_input as eff_pre
8
 
9
  # Load models
10
+ xcp_path = hf_hub_download(repo_id="Zeyadd-Mostaffa/deepfake-image-detector_final", filename="xception_model.h5")
11
+ eff_path = hf_hub_download(repo_id="Zeyadd-Mostaffa/deepfake-image-detector_final", filename="efficientnet_model.h5")
12
+ xcp_model = load_model(xcp_path)
13
+ eff_model = load_model(eff_path)
14
+
15
 
16
  # Load face detector
17
  detector = MTCNN()