Fixed RGB

app.py

@@ -1,4 +1,5 @@
 import os
+import cv2
 import gradio as gr
 from ultralytics import YOLO
 import numpy as np
@@ -29,7 +30,8 @@ def process_image(input_image, model_name, conf):
     print('model_index: ')
     print(model_index)
     model = models[model_index]
-    
+    input_image = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
+#    cv2.imwrite('./input_image.jpg', input_image)    
     results = model.predict(input_image, conf=conf)
     class_counts = {}
     class_counts_str = "Class Counts:\n"
@@ -37,6 +39,7 @@ def process_image(input_image, model_name, conf):
     for r in results:
         im_array = r.plot()
         im_array = im_array.astype(np.uint8)
+        im_array = cv2.cvtColor(im_array, cv2.COLOR_BGR2RGB)
 
         if len(r.boxes) == 0:  # If no objects are detected
             return None, "No objects detected."