Spaces:

linhcuem
/

cham_diem_yolov8

Sleeping

linhcuem commited on Oct 23, 2023

Commit

3dfb48a

1 Parent(s): 7ebebbd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,7 +56,7 @@ def yolov8_img_inference(
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
     # count_result = results[0].boxes.cls.tolist()
-    clist= res[0].boxes.cls
     cls = set()
     for cno in clist:
         cls.add(model.names[int(cno)])

     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
     # count_result = results[0].boxes.cls.tolist()
+    clist= results[0].boxes.cls
     cls = set()
     for cno in clist:
         cls.add(model.names[int(cno)])