Spaces:

nabeelraza
/

wheat-disease-detection-XAI

Sleeping

nabeelraza commited on Apr 15

Commit

2812dd4

verified ·

1 Parent(s): 5a399e2

Update explain.py

Files changed (1) hide show

explain.py CHANGED Viewed

@@ -21,8 +21,8 @@ import uuid
 # Static variables
-model_path = "efficientnet-b0-best.pth"
-model_name = "efficientnet_b0"
 YOLO_MODEL_WEIGHTS = "yolo-v11-best.pt"
 classes = ["Healthy", "Resistant", "Susceptible"]
 resizing_transforms = transforms.Compose([transforms.CenterCrop(224)])

 # Static variables
+model_path = "efficientnet-b2.pth"
+model_name = "efficientnet_b2"
 YOLO_MODEL_WEIGHTS = "yolo-v11-best.pt"
 classes = ["Healthy", "Resistant", "Susceptible"]
 resizing_transforms = transforms.Compose([transforms.CenterCrop(224)])