Zeyadd-Mostaffa commited on
Commit
a2e411b
·
verified ·
1 Parent(s): d888b34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -8,6 +8,7 @@ from tensorflow.keras.applications.xception import preprocess_input as xcp_pre
8
  from tensorflow.keras.applications.efficientnet import preprocess_input as eff_pre
9
  from tensorflow.keras.preprocessing.image import img_to_array
10
  from huggingface_hub import hf_hub_download
 
11
 
12
  # Load models
13
  xcp_path = hf_hub_download(repo_id="Zeyadd-Mostaffa/deepfake-image-detector_final", filename="xception_model.h5")
@@ -61,3 +62,14 @@ def predict_image(image_path):
61
  results.append({"face_id": f"face_{idx+1}", "label": label, "score": round(float(final_score), 3)})
62
 
63
  return results
 
 
 
 
 
 
 
 
 
 
 
 
8
  from tensorflow.keras.applications.efficientnet import preprocess_input as eff_pre
9
  from tensorflow.keras.preprocessing.image import img_to_array
10
  from huggingface_hub import hf_hub_download
11
+ import gradio as gr
12
 
13
  # Load models
14
  xcp_path = hf_hub_download(repo_id="Zeyadd-Mostaffa/deepfake-image-detector_final", filename="xception_model.h5")
 
62
  results.append({"face_id": f"face_{idx+1}", "label": label, "score": round(float(final_score), 3)})
63
 
64
  return results
65
+
66
+ interface = gr.Interface(
67
+ fn=predict_image, # your prediction function
68
+ inputs=gr.Image(type="filepath"),
69
+ outputs="json",
70
+ title="Deepfake Detector (Hybrid Strategy)",
71
+ description="If 1 face → predict full image. If >1 → predict each face."
72
+ )
73
+
74
+ # ✅ Required to start app
75
+ interface.launch()