Hope-Liang commited on
Commit
b0a9f12
·
1 Parent(s): 8957a74
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -83,6 +83,7 @@ model_dir = model.download()
83
  model = joblib.load(model_dir + "/incident_model.pkl")
84
 
85
  batch_data = results_df_preprocessed.copy()
 
86
  batch_data.drop(columns=['incident_datetime','incident_category'], inplace=True)
87
  y_pred = model.predict(batch_data)
88
 
 
83
  model = joblib.load(model_dir + "/incident_model.pkl")
84
 
85
  batch_data = results_df_preprocessed.copy()
86
+ batch_data = batch_data[:100]
87
  batch_data.drop(columns=['incident_datetime','incident_category'], inplace=True)
88
  y_pred = model.predict(batch_data)
89