Update app.py

app.py

@@ -42,7 +42,7 @@ if uploaded_file is not None:
     predicted_class = class_scores.argmax()
     
 
-    classes = ['Low Quality', 'Medium Quality', 'High Quality']  # Adjust according to your classes
+    classes = ['Low Quality', 'Medium Quality', 'High Quality'] 
     st.write(f"**Prediction:** {classes[predicted_class]}")
     st.write(f"**Confidence:** {np.max(class_scores) * 100:.2f}%")