Update app.py

app.py

@@ -163,9 +163,15 @@ def main():
                 # Predict and display results
                 if st.sidebar.button("Predict"):
                     predictions = rf_model.predict(input_data)
-                    prediction_columns = ["Predicted Change in PTS", "Predicted Change in REB", "Predicted Change inAST"]
+                    # 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"]
+                    
                     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.")