eeshawn11 commited on
Commit
691c828
·
1 Parent(s): da44c62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -22,7 +22,8 @@ def yolov8_inference(
22
  model = YOLO("ultralyticsplus/yolov8s")
23
  model.conf = conf_threshold
24
  model.iou = iou_threshold
25
- results = model.predict(image, return_outputs=True)
 
26
  object_prediction_list = []
27
  annotator = Annotator(image)
28
  for _, image_results in enumerate(results):
 
22
  model = YOLO("ultralyticsplus/yolov8s")
23
  model.conf = conf_threshold
24
  model.iou = iou_threshold
25
+ # results = model.predict(image, return_outputs=True)
26
+ results = model.predict(image)
27
  object_prediction_list = []
28
  annotator = Annotator(image)
29
  for _, image_results in enumerate(results):