muskan19 commited on
Commit
7391c3b
·
verified ·
1 Parent(s): a42185d

Update src/predict.py

Browse files
Files changed (1) hide show
  1. src/predict.py +2 -2
src/predict.py CHANGED
@@ -1,10 +1,10 @@
1
- # predict.py placeholder
2
  from tensorflow.keras.models import load_model
3
 
4
  def load_trained_model(path):
5
  return load_model(path)
6
 
7
- def predict_violence(model, processed_frame):
8
  prediction = model.predict(processed_frame)
9
  return prediction[0][0]
10
 
 
 
 
1
  from tensorflow.keras.models import load_model
2
 
3
  def load_trained_model(path):
4
  return load_model(path)
5
 
6
+ def run_prediction(model, processed_frame): # Renamed function
7
  prediction = model.predict(processed_frame)
8
  return prediction[0][0]
9
 
10
+