Update src/predict.py

src/predict.py

@@ -2,7 +2,7 @@ from tensorflow.keras.models import load_model
 
 #def load_trained_model(path):
     #return load_model(path)
-
+'''
 import os
 import urllib.request
 from tensorflow.keras.models import load_model
@@ -16,10 +16,27 @@ def load_trained_model(path="violence_model.h5"):
         print("Model downloaded.")
 
     return load_model(path, compile=False)
-
+'''
 
 def run_prediction(model, processed_frame):   # Renamed function
     prediction = model.predict(processed_frame)
     return prediction[0][0]
 
 
+
+import os
+import urllib.request
+from tensorflow.keras.models import load_model
+
+def load_trained_model(path="violence_model.h5"):
+    url = "https://huggingface.co/spaces/muskan19/Violence_Detector/resolve/main/violence_model.h5"
+
+    if not os.path.exists(path):
+        print("Downloading model...")
+        urllib.request.urlretrieve(url, path)
+        print("Download complete.")
+
+    return load_model(path, compile=False)
+
+
+