Update src/predict.py

src/predict.py

@@ -72,7 +72,7 @@ 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
+        filename="violence_model.h5"     # Exact file name
     )
     return load_model(model_path, compile=False)