Update app.py

app.py

@@ -1,31 +1,32 @@
 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 ultralyticsplus import YOLO
 from PIL import Image
 
-# Load YOLOv8 leaf detection model
+# 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 PIL Image if needed
+    # Convert to a PIL Image (if not already)
     image = Image.open(image).convert("RGB")
-
     # Run inference
     results = model.predict(image)
-
-    # Count number of detected leaves
+    # Count the number of detected leaves
     num_leaves = len(results[0].boxes)
-
     return f"Number of leaves detected: {num_leaves}"
 
 # Gradio UI
 iface = gr.Interface(
     fn=count_leaves,
-    inputs=gr.Image(type="filepath"),  # User uploads an image
+    inputs=gr.Image(type="filepath"),
     outputs="text",
     title="Leaf Counter",
     description="Upload an image of a plant, and the model will detect and count the number of leaves."
 )
 
-# Launch app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()