Update app.py

app.py

@@ -50,18 +50,29 @@ def load_text_model():
 # ---------------------------
 # LOAD IMAGE MODEL (PyTorch only, no TensorFlow)
 # ---------------------------
+# @st.cache_resource
+# def load_image_model():
+#     import os
+#     import tensorflow as tf
+#     model_path = "image_model.h5"
+
+#     if not os.path.exists(model_path):
+#         url = "https://huggingface.co/syeda-Rija20/image-detector/resolve/main/image_detector_finetuned.h5"
+#         response = requests.get(url)
+#         with open(model_path, "wb") as f:
+#             f.write(response.content)
+
+#     model = tf.keras.models.load_model(model_path)
+#     return model
 @st.cache_resource
 def load_image_model():
-    import os
-    import tensorflow as tf
-    model_path = "image_model.h5"
-
-    if not os.path.exists(model_path):
-        url = "https://huggingface.co/syeda-Rija20/image-detector/resolve/main/image_detector_finetuned.h5"
-        response = requests.get(url)
-        with open(model_path, "wb") as f:
-            f.write(response.content)
+    import os, tensorflow as tf
+    from huggingface_hub import hf_hub_download
 
+    model_path = hf_hub_download(
+        repo_id="Muniba930/image-detector",
+        filename="image_detector_v2.h5"
+    )
     model = tf.keras.models.load_model(model_path)
     return model