abdrabo01 commited on
Commit
52b3dfe
·
verified ·
1 Parent(s): f66d891

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -48,7 +48,7 @@ async def predict(request: ImageRequest):
48
  json_result = {}
49
  class_counters = {}
50
 
51
- for box in result.boxes:
52
  class_id = int(box.cls[0])
53
  class_name = result.names[class_id]
54
  x1, y1, x2, y2 = map(int, box.xyxy[0].tolist())
 
48
  json_result = {}
49
  class_counters = {}
50
 
51
+ for box in result.boxes:
52
  class_id = int(box.cls[0])
53
  class_name = result.names[class_id]
54
  x1, y1, x2, y2 = map(int, box.xyxy[0].tolist())