Update app.py

app.py

@@ -66,13 +66,31 @@ def load_text_model():
 #     return model
 @st.cache_resource
 def load_image_model():
-    import os, tensorflow as tf
-    from huggingface_hub import hf_hub_download
+    import os
+    import tensorflow as tf
+
+    model_path = "/tmp/image_detector_v2.h5"
+
+    if not os.path.exists(model_path):
+        import requests
+        print("Downloading model...")
+        url = "https://huggingface.co/Muniba930/image-detector/resolve/main/image_detector_v2.h5?download=true"
+        
+        with requests.get(url, stream=True) as r:
+            r.raise_for_status()
+            with open(model_path, "wb") as f:
+                for chunk in r.iter_content(chunk_size=8192):
+                    f.write(chunk)
+        
+        # Verify file size
+        size_mb = os.path.getsize(model_path) / (1024 * 1024)
+        print(f"Downloaded: {size_mb:.1f} MB")
+        
+        if size_mb < 10:
+            os.remove(model_path)
+            st.error(f"Download failed — only {size_mb:.1f} MB received")
+            return None
 
-    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