Update app.py

app.py

@@ -1,9 +1,6 @@
 import gradio as gr
 from ultralytics import YOLO
 import spaces
-import supervision as sv
-
-box_annotator = sv.BoxAnnotator()
 
 category_dict = {
     0: 'person', 1: 'bicycle', 2: 'car', 3: 'motorcycle', 4: 'airplane', 5: 'bus',
@@ -50,6 +47,7 @@ def LeYOLO_inference(image, model_id, image_size, conf_threshold, iou_threshold)
     model = model = YOLO(MODEL_PATH)
     results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
     annotated_image = results[0].plot()
+    annotated_image = cv2.cvtColor(annotated_image, cv2.COLOR_RGB2BGR)
     return annotated_image