Update app.py

app.py

@@ -58,11 +58,11 @@ def predict_image(input_image, name, age, medical_record, sex):
         # Determine the label based on the class index
         class_index = highest_confidence_result.cls.item()
         if class_index == 0:
-            label = "Mature"
-            color = (255, 0, 0)  # Red for Mature
-        else:
             label = "Normal"
             color = (0, 255, 0)  # Green for Normal
+        elif class_index == 1:
+            label = "Cataract"
+            color = (255, 0, 0)  # Red for Cataract
 
         confidence = highest_confidence_result.conf.item()
         xmin, ymin, xmax, ymax = map(int, highest_confidence_result.xyxy[0])