Update app.py

app.py

@@ -3,9 +3,10 @@ import torch
 from ultralytics.nn.tasks import DetectionModel
 from torch.nn.modules.container import Sequential
 from ultralytics.nn.modules import Conv
+from torch.nn.modules.conv import Conv2d  # Import Conv2d
 
-# Whitelist safe globals (only do this if you trust the source of the model)
-torch.serialization.add_safe_globals([DetectionModel, Sequential, Conv])
+# Whitelist the globals to bypass the pickle errors (only do this if you trust the source!)
+torch.serialization.add_safe_globals([DetectionModel, Sequential, Conv, Conv2d])
 
 from ultralyticsplus import YOLO, render_result
 from PIL import Image
@@ -13,7 +14,7 @@ from PIL import Image
 # Load the YOLOv8s Leaf Detection and Classification model from Hugging Face
 model = YOLO('foduucom/plant-leaf-detection-and-classification')
 
-# Set recommended model parameters
+# Set recommended model parameters as per the model card
 model.overrides['conf'] = 0.25       # NMS confidence threshold
 model.overrides['iou'] = 0.45        # NMS IoU threshold
 model.overrides['agnostic_nms'] = False  # NMS class-agnostic setting