alibidaran commited on
Commit
c113153
·
verified ·
1 Parent(s): 9058f9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def detect_objects(image):
19
  for box,cls in zip(boxes,classes):
20
  annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
21
  annotator1.box_label(box, label=None, color=colors(int(cls)))
22
- return Image.fromarray(annotator.result()),Image.fromarray(annotator1.result())
23
 
24
  # Gradio Interface
25
  title = "YOLOv8 Object Detection"
 
19
  for box,cls in zip(boxes,classes):
20
  annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
21
  annotator1.box_label(box, label=None, color=colors(int(cls)))
22
+ return Image.fromarray(annotator.result()),Image.fromarray(annotator1.result()),None
23
 
24
  # Gradio Interface
25
  title = "YOLOv8 Object Detection"