Update app.py

app.py

@@ -26,54 +26,54 @@ class MMIM(nn.Module):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = MMIM(num_classes=36)
 
+# 🧠 Load only matching weights from checkpoint (skip classifier mismatch)
 checkpoint = torch.load("MMIM_best.pth", map_location=device)
 filtered_checkpoint = {
-    k: v for k, v in checkpoint.items()
-    if k in model.state_dict() and model.state_dict()[k].shape == v.shape
+    k: v for k, v in checkpoint.items() if k in model.state_dict() and model.state_dict()[k].shape == v.shape
 }
 model.load_state_dict(filtered_checkpoint, strict=False)
+
 model.to(device)
 model.eval()
 
-# ✅ Final class_names (corrected index order, without class10–class13)
+# ✅ class_names mapped according to confusion matrix order
 class_names = [
-    "Chinee apple",              # 0 = class1
-    "Lantana",                   # 1 = class2
-    "Negative",                  # 2 = class3
-    "Parkinsonia",              # 3 = class4
-    "Parthenium",               # 4 = class5
-    "Prickly acacia",           # 5 = class6
-    "Rubber vine",              # 6 = class7
-    "Siam weed",                # 7 = class8
-    "Snake weed",               # 8 = class9
-    # Skipping class10–class13
-    "Common Wheat",             # 9 = class14
-    "Fat Hen",                  # 10 = class15
-    "Loose Silky-bent",         # 11 = class16
-    "Maize",                    # 12 = class17
-    "Scentless Mayweed",        # 13 = class18
-    "Shepherds purse",          # 14 = class19
-    "Small-flowered Cranesbill",# 15 = class20
-    "Sugar beet",               # 16 = class21
-    "Carpetweeds",              # 17 = class22
-    "Crabgrass",                # 18 = class23
-    "Eclipta",                  # 19 = class24
-    "Goosegrass",               # 20 = class25
-    "Morningglory",             # 21 = class26
-    "Nutsedge",                 # 22 = class27
-    "PalmerAmaranth",           # 23 = class28
-    "Pricky Sida",              # 24 = class29
-    "Purslane",                 # 25 = class30
-    "Ragweed",                  # 26 = class31
-    "Sicklepod",                # 27 = class32
-    "SpottedSpurge",            # 28 = class33
-    "Spurred Anoda",            # 29 = class34
-    "Swinecress",               # 30 = class35
-    "Waterhemp",                # 31 = class36
-    "Extra1",                   # 32 = class37 (if any)
-    "Extra2",                   # 33 = class38
-    "Extra3",                   # 34 = class39
-    "Extra4"                    # 35 = class40
+    "Chinee apple",              # class1
+    "Black grass",                   # class2
+    "Negative",                  # class3
+     "Rubber vine",               # class7
+     "Snake weed",                # class9
+     "Black grass",               # class10
+     "Charlock",                  # class11
+      "Cleavers",                  # class12
+      "Common Chickweed"           # class13
+      "Common Wheat",              # class14
+    "Fat Hen",                   # class15
+     "Loose Silky-bent",          # class16
+    "Maize",                     # class17
+    "Scentless Mayweed",         # class18
+    "Shepherds purse",           # class19
+    "Small-flowered Cranesbill",# class20
+    "Sugar beet",                # class21
+    "Carpetweeds",               # class22
+    "Crabgrass",                 # class23
+    "Eclipta",                   # class24
+    "Goosegrass",                # class25
+    "Morningglory",              # class26
+    "Nutsedge",                  # class27
+    "PalmerAmaranth",            # class28
+    "Pricky Sida",               # class29
+    "Purslane",                  # class30
+    "Ragweed",                   # class31
+    "Sicklepod",                 # class32
+    "SpottedSpurge",             # class33
+    "SpurredAnoda",              # class34
+    "Swinecress",                # class35
+    "Waterhemp",                 # class36
+    "Parkinsonia",               # class4
+    "Parthenium",                # class5
+    "Prickly acacia",            # class6
+   "Siam weed",                 # class8
 ]
 
 # 🔁 Image transform
@@ -109,4 +109,4 @@ interface = gr.Interface(
     description="Upload a weed image to predict its class. If the model detects a non-weed image, it will return 'Negative'."
 )
 
-interface.launch()
+interface.launch()