Update app.py

app.py

@@ -36,8 +36,8 @@ if st.button("Predict"):
         #predictions_np = np.array(predictions)
         
         # Scale the predictions
-        scaled_scores = 2.25 * predicted_scores - 1.25
-        rounded_scores = [round(score * 2) / 2 for score in scaled_scores]  # Round to nearest 0.5
+        normalized_scores = (predicted_scores / predicted_scores.max()) * 9  # Scale to 9
+        rounded_scores = np.round(normalized_scores * 2) / 2
 
         # Display the predictions
         labels = ["Task Achievement", "Coherence and Cohesion", "Vocabulary", "Grammar", "Overall"]