Update app.py

app.py

@@ -36,16 +36,44 @@ model.load_state_dict(filtered_checkpoint, strict=False)
 model.to(device)
 model.eval()
 
-# ✅ Correct class names list (ordered by folder names)
+# ✅ class_names mapped according to confusion matrix order
 class_names = [
-    "Chinee apple", "Lantana", "Negative", "Parkinsonia", "Parthenium",
-    "Prickly acacia", "Rubber vine", "Siam weed", "Snake weed", 
-    "Black grass", "Charlock", "Cleavers", "Common Chickweed", "Common Wheat",
-    "Fat Hen", "Loose Silky-bent", "Maize", "Scentless Mayweed",
-    "Shepherds purse", "Small-flowered Cranesbill", "Sugar beet",
-    "Carpetweeds", "Crabgrass", "Eclipta", "Goosegrass", "Morningglory",
-    "Nutsedge", "PalmerAmaranth", "Pricky Sida", "Purslane", "Ragweed",
-    "Sicklepod", "SpottedSpurge", "SpurredAnoda", "Swinecress", "Waterhemp"
+    "Chinee apple",              # class1
+    "Common Wheat",              # class14
+    "Fat Hen",                   # class15
+    "Loose Silky-bent",          # class16
+    "Maize",                     # class17
+    "Scentless Mayweed",         # class18
+    "Shepherds purse",           # class19
+    "Lantana",                   # class2
+    "Small-flowered Cranesbill",# class20
+    "Sugar beet",                # class21
+    "Carpetweeds",               # class22
+    "Crabgrass",                 # class23
+    "Eclipta",                   # class24
+    "Goosegrass",                # class25
+    "Morningglory",              # class26
+    "Nutsedge",                  # class27
+    "PalmerAmaranth",            # class28
+    "Pricky Sida",               # class29
+    "Negative",                  # class3
+    "Purslane",                  # class30
+    "Ragweed",                   # class31
+    "Sicklepod",                 # class32
+    "SpottedSpurge",             # class33
+    "SpurredAnoda",              # class34
+    "Swinecress",                # class35
+    "Waterhemp",                 # class36
+    "Parkinsonia",               # class4
+    "Parthenium",                # class5
+    "Prickly acacia",            # class6
+    "Rubber vine",               # class7
+    "Siam weed",                 # class8
+    "Snake weed",                # class9
+    "Black grass",               # class10
+    "Charlock",                  # class11
+    "Cleavers",                  # class12
+    "Common Chickweed"           # class13
 ]
 
 # 🔁 Image transform
@@ -70,7 +98,7 @@ def predict(img):
     if predicted_class.lower() == "negative":
         return f"⚠️ This image is predicted as Negative.\nConfidence: {confidence:.2f}%"
 
-    return f"✅ Predicted as a weed with class-{predicted_class}\nConfidence: {confidence:.2f}%"
+    return f"✅ Predicted class: {predicted_class}\nConfidence: {confidence:.2f}%"
 
 # 🎨 Gradio Interface
 interface = gr.Interface(