kedimestan commited on
Commit
102dcbc
·
verified ·
1 Parent(s): 7f01571

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -56,12 +56,12 @@ def yolov8_inference(
56
  model.overrides["iou"] = iou_threshold
57
  results = model.predict(image, imgsz=image_size, return_outputs=True)
58
 
59
- boxes = []
60
  for result in results:
61
  # Extract bounding boxes (xyxy format)
62
- boxes.append(result["boxes"].cpu().numpy())
63
 
64
- return boxes
65
 
66
 
67
  inputs = [
 
56
  model.overrides["iou"] = iou_threshold
57
  results = model.predict(image, imgsz=image_size, return_outputs=True)
58
 
59
+ boxes1 = []
60
  for result in results:
61
  # Extract bounding boxes (xyxy format)
62
+ boxes1.append(result.boxes.cpu().numpy())
63
 
64
+ return boxes1
65
 
66
 
67
  inputs = [