Update src/predict.py

src/predict.py

@@ -39,6 +39,7 @@ def load_trained_model(path="violence_model.keras"):
 
     return load_model(path, compile=False)
 '''
+'''
 from huggingface_hub import hf_hub_download
 from tensorflow.keras.models import load_model
 
@@ -48,6 +49,19 @@ def load_trained_model():
         filename="violence_model.keras"
     )
     return load_model(model_path, compile=False)
+'''
+
+import os
+import urllib.request
+from tensorflow.keras.models import load_model
+
+def load_trained_model(path="violence_model.keras"):
+    url = "https://huggingface.co/spaces/muskan19/Violence_Detector/resolve/main/violence_model.keras"
+    if not os.path.exists(path):
+        print("📥 Downloading model...")
+        urllib.request.urlretrieve(url, path)
+    return load_model(path, compile=False)
+