Update app.py

app.py

@@ -13,14 +13,14 @@ def make_prediction(model, input_data):
     return prediction
 
 # Define the function to calculate GPA
-def calculate_gpa(score):
-    if score >= 70:
+def calculate_gpa(total_score):
+    if total_score >= 70:
         return 'A (5 points)'
-    elif score >= 60:
+    elif total_score >= 60:
         return 'B (4 points)'
-    elif score >= 50:
+    elif total_score >= 50:
         return 'C (3 points)'
-    elif score >= 45:
+    elif total_score >= 45:
         return 'D (2 points)'
     else:
         return 'F (0 points)'
@@ -46,16 +46,16 @@ def main():
         mid_semester = st.slider("Mid Semester Score", min_value=1, max_value=20, step=1)
         assignments = st.slider("Assignments", min_value=1, max_value=10, step=1)
 
-    # Create input data
-    input_data = np.array([[level, course_units, attendance, mid_semester, assignments]])
+    # Calculate total score
+    total_score = attendance + mid_semester + assignments
 
-    # Make predictionS
+    # Make prediction
     if st.button("Predict Exam Score"):
-        prediction = make_prediction(model, input_data)
+        prediction = make_prediction(model, np.array([[level, course_units, total_score]]))
         st.write(f"Predicted Exam Score: {prediction[0]:.2f}")
-        
+
         # Calculate GPA
-        gpa = calculate_gpa(prediction[0])
+        gpa = calculate_gpa(total_score)
         st.write(f"Predicted GPA: {gpa}")
 
 if __name__ == '__main__':