hot_fix

app.py

@@ -100,11 +100,11 @@ content = "".join(content)
 classes = content.split("\n")
 color_palette_pred = np.random.uniform(0, 255, size=(len(classes), 3))  
   
-model_names = ["yolov8n 640 mask_300000_3000_0.3_0.7.onnx",
-               "yolov8s 640 mask_300000_3000_0.3_0.7.onnx",
-               "yolov8m 640 mask_300000_3000_0.3_0.7.onnx",
-               "yolov8l 640 mask_300000_3000_0.3_0.7.onnx",
-               "yolov8x 640 mask_300000_3000_0.3_0.7.onnx"]
+model_names = {"YOLOv8n":"yolov8n 640 mask_300000_3000_0.3_0.7.onnx",
+               "YOLOv8s":"yolov8s 640 mask_300000_3000_0.3_0.7.onnx",
+               "YOLOv8m":"yolov8m 640 mask_300000_3000_0.3_0.7.onnx",
+               "YOLOv8l":"yolov8l 640 mask_300000_3000_0.3_0.7.onnx",
+               "YOLOv8x":"yolov8x 640 mask_300000_3000_0.3_0.7.onnx"}
 input_size = 640