Update app.py

app.py

@@ -163,15 +163,10 @@ def main():
                 # Predict and display results
                 if st.sidebar.button("Predict"):
                     predictions = rf_model.predict(input_data)
-                    # Round predictions to two decimal places
                     predictions = predictions.round(2)
-                    
-                    # Define column names for the output table
-                    prediction_columns = ["Predicted Change in PTS", "Predicted Change in REB", "Predicted Change in AST"]
-                    
+                    prediction_columns = ["Predicted Change in PTS", "Predicted Change in REB", "Predicted Change inAST"]
                     st.subheader("Predicted Post-Injury Performance")
                     st.write("Based on the inputs, here are the predicted metrics:")
-                    st.table(pd.DataFrame([predictions], columns=prediction_columns))
                     st.table(pd.DataFrame(predictions, columns=prediction_columns))
             except FileNotFoundError:
                 st.error("Model file not found.")