Update app.py

app.py

@@ -57,7 +57,7 @@ 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 == 1:
+        if class_index == 0:
             label = "Mature"
             color = (255, 0, 0)  # Red for Mature
         else: