Zakia commited on
Commit
cba1611
1 Parent(s): cb67d6b

remove [0]

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = load_model("tuned_blend_specific_model_19112021")
15
  # define the function to call
16
  def predict(model, input_df):
17
  predictions_df = predict_model(estimator=model, data=input_df)
18
- predictions = predictions_df["AMPUTATION"][0]
19
 
20
 
21
 
 
15
  # define the function to call
16
  def predict(model, input_df):
17
  predictions_df = predict_model(estimator=model, data=input_df)
18
+ predictions = predictions_df["AMPUTATION"]
19
 
20
 
21