Spaces:

linhcuem
/

cham_diem_yolov8

Sleeping

linhcuem commited on Oct 24, 2023

Commit

1a1ec2d

•

1 Parent(s): cb31023

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ def yolov8_img_inference(
     clist = results[0].boxes.cls
     cls = set()
     for c in clist:
-        cls = cls.add(names[int(c)])
         if c in names:
             object_counts[c] += 1
         elif c not in names:

     clist = results[0].boxes.cls
     cls = set()
     for c in clist:
+        cls.add(names[int(c)])
         if c in names:
             object_counts[c] += 1
         elif c not in names: