App bug fix.

app.py

@@ -83,9 +83,19 @@ model_yolos.to(device)
 model_yolos.eval()
 
 # colors for visualization
-colors = ['Pink', 'Crimson', 'Yellow', 'Indigo', 'BlueViolet',
-          'Blue', 'GhostWhite', 'LightSteelBlue', 'Brown', 'SkyBlue',
-          'Tomato']
+COLORS = [
+    [0.000, 0.447, 0.741], 
+    [0.850, 0.325, 0.098], 
+    [0.929, 0.694, 0.125],
+    [0.756, 0.794, 0.100], 
+    [0.466, 0.674, 0.188], 
+    [0.301, 0.745, 0.933],
+    [0.184, 0.494, 0.741], 
+    [0.494, 0.674, 0.556],
+     [0.494, 0.301, 0.933],
+    [0.000, 0.325, 0.850], 
+    [0.745, 0.301, 0.188]]
+
 # for output bounding box post-processing
 def box_cxcywh_to_xyxy(x):
     x_c, y_c, w, h = x.unbind(1)
@@ -108,7 +118,7 @@ def plot_results(pil_img, prob, boxes):
         c = colors[cl]
         c1, c2 = (int(xmin), int(ymin)), (int(xmax), int(ymax))
 
-        cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
+        cv2.rectangle(img, c1, c2, [0,255,255], thickness=2, lineType=cv2.LINE_AA)
         # cv2.text(
         #     [xmin + 5, ymin + 5], 
         #     f'{id2label[cl.item()]}: {p[cl]:0.2f}',