lrschuman17 commited on
Commit
5663fe1
·
verified ·
1 Parent(s): f707102

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -163,7 +163,7 @@ def main():
163
  # Predict and display results
164
  if st.sidebar.button("Predict"):
165
  predictions = rf_model.predict(input_data)
166
- predictions = predictions.round(2)
167
  prediction_columns = ["Predicted Change in PTS", "Predicted Change in REB", "Predicted Change inAST"]
168
  st.subheader("Predicted Post-Injury Performance")
169
  st.write("Based on the inputs, here are the predicted metrics:")
 
163
  # Predict and display results
164
  if st.sidebar.button("Predict"):
165
  predictions = rf_model.predict(input_data)
166
+ predictions = [round(float(pred), 2) for pred in predictions]
167
  prediction_columns = ["Predicted Change in PTS", "Predicted Change in REB", "Predicted Change inAST"]
168
  st.subheader("Predicted Post-Injury Performance")
169
  st.write("Based on the inputs, here are the predicted metrics:")