Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import pickle
-from sklearn.ensemble import RandomForestRegressor
 import numpy as np
 
 # Load the saved model
@@ -51,7 +50,9 @@ def main():
 
     # Make prediction
     if st.button("Predict Exam Score"):
-        prediction = make_prediction(model, np.array([[level, course_units, total_score]]))
+        # Add total_score to the input data
+        input_data = np.array([[level, course_units, attendance, mid_semester, assignments]])
+        prediction = make_prediction(model, input_data)
         st.write(f"Predicted Exam Score: {prediction[0]:.2f}")
 
         # Calculate GPA