nehulagrawal commited on
Commit
d9c2732
·
1 Parent(s): fd44d26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -12,14 +12,15 @@ def get_predictions(img, threshold, box_color, text_color):
12
  v8_frame = yolov8_detector.plot_bboxes(v8_results, img, float(threshold), box_color, text_color)
13
  return v8_frame
14
 
15
- # Load the YOLOv8 model for plant leaf detection and classification
16
- yolov8_detector = ObjectDetection('Yolov8')
17
 
18
- with gr.Blocks(title="Plant Leaf Detection and Classification", theme=gr.themes.DarkMode()) as interface:
19
- # Add a header with a description of the app and the model used
20
  gr.Markdown("# Plant Leaf Detection and Classification")
21
- gr.Markdown("This app uses YOLOv8, a state-of-the-art object detection model, to detect and classify plant leaves. "
22
- "The model can detect leaves and classify them into up to 45 different plant categories.")
 
 
23
 
24
  with gr.Row():
25
  with gr.Column():
 
12
  v8_frame = yolov8_detector.plot_bboxes(v8_results, img, float(threshold), box_color, text_color)
13
  return v8_frame
14
 
15
+ # Load the YOLOv8s model for plant leaf detection and classification
16
+ yolov8_detector = ObjectDetection('Yolov8s')
17
 
18
+ with gr.Blocks(title="Plant Leaf Detection and Classification", theme=gr.themes.Soft()) as interface:
 
19
  gr.Markdown("# Plant Leaf Detection and Classification")
20
+ gr.Markdown("This app uses YOLOv8s, a powerful object detection model, to detect and classify plant leaves. "
21
+ "The model can detect various plant leaves and classify them into up to 45 different plant categories, "
22
+ "indicating whether they are diseased or healthy.")
23
+ gr.Markdown("If you like this app, don't forget to give it a thumbs up on Hugging Face!")
24
 
25
  with gr.Row():
26
  with gr.Column():