Update src/predict.py

src/predict.py

@@ -50,7 +50,7 @@ def load_trained_model():
     )
     return load_model(model_path, compile=False)
 '''
-
+'''
 import os
 import urllib.request
 from tensorflow.keras.models import load_model
@@ -61,6 +61,17 @@ def load_trained_model(path="violence_model.keras"):
         print("📥 Downloading model...")
         urllib.request.urlretrieve(url, path)
     return load_model(path, compile=False)
+'''
+
+from huggingface_hub import hf_hub_download
+from tensorflow.keras.models import load_model
+
+def load_trained_model():
+    model_path = hf_hub_download(
+        repo_id="muskan19/violence_model",  # Your model repo, NOT Space
+        filename="violence_model.keras"     # Exact file name
+    )
+    return load_model(model_path, compile=False