Update app.py

app.py

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