Update app.py

app.py

@@ -1,21 +1,21 @@
 import gradio as gr
 import torch
-from ultralytics.nn.tasks import DetectionModel
-# Allow safe globals for custom model loading (do this only if you trust the source)
-torch.serialization.add_safe_globals([DetectionModel])
+from torch.nn.modules.container import Sequential  # import Sequential
+# Allow safe globals for the Sequential class
+torch.serialization.add_safe_globals([Sequential])
+
 from ultralyticsplus import YOLO
 from PIL import Image
 
 # Load your custom YOLOv8 leaf detection model.
-# If this still causes issues, try using a supported model like 'yolov8n.pt'
 model = YOLO('foduucom/plant-leaf-detection-and-classification')
 
 def count_leaves(image):
-    # Convert to a PIL Image (if not already)
+    # Ensure the image is a PIL Image and convert to RGB
     image = Image.open(image).convert("RGB")
     # Run inference
     results = model.predict(image)
-    # Count the number of detected leaves
+    # Count detected leaves
     num_leaves = len(results[0].boxes)
     return f"Number of leaves detected: {num_leaves}"