SarowarSaurav commited on
Commit
744ed2f
·
verified ·
1 Parent(s): 279c1fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,9 +4,9 @@ from PIL import Image
4
  import numpy as np
5
  import cv2
6
 
7
- # Load the pre-trained YOLOv5 model
8
- # Use 'yolov5s' or any available YOLO model from ultralytics
9
- model = YOLO('yolov5s') # Change to a leaf disease detection model if available
10
 
11
  def identify_disease(image):
12
  # Convert the image to RGB if it's not
 
4
  import numpy as np
5
  import cv2
6
 
7
+ # This should download the model if not available locally
8
+ model = YOLO('yolov8n.pt') # Using a YOLOv8 model which is often smaller and compatible
9
+ # Change to a leaf disease detection model if available
10
 
11
  def identify_disease(image):
12
  # Convert the image to RGB if it's not