Update app.py

app.py

@@ -22,9 +22,9 @@ def predict(image):
     # Map the predicted class index to pain levels
     pain_levels = {
         0: "No Pain",
-        1: "Mild Pain",
-        2: "Moderate Pain",
-        3: "Severe Pain",
+        1: "Low Pain",
+        2: "Medium Pain",
+        3: "High Pain",
     }
     
     return pain_levels[predicted_class[0]]  # Return the corresponding pain level