Fixed exception

app.py

@@ -113,7 +113,7 @@ def detect(img,model,device,iou_threshold=0.45,confidence_threshold=0.25):
             # Write results
             for *xyxy, conf, cls in reversed(det):
                 label = f'{names[int(cls)]} {conf:.2f}'
-                plot_one_box(xyxy, imgs, label=label, color=colors[int(cls)], line_thickness=2)
+                plot_one_box(xyxy, imgs, label=label, color=colors[names[int(cls)]], line_thickness=2)
 
     return imgs,fps_inference