fix channels

app.py

@@ -26,12 +26,15 @@ def load_model():
     return yolo
 
 def detect(image: np.ndarray, detector: YOLO) -> np.ndarray:
+    # input is rgb image, output must be rgb too
+
     detector_kwargs = {'conf': 0.5, 'iou': 0.5, 'half': False, 'verbose': False}
 
+    image = image[:, :, ::-1]  # RGB -> BGR
     results: Results = detector.predict(image, **detector_kwargs)[0]
     out_im = results.plot()
 
-    return out_im
+    return out_im[:, :, ::-1]  # BGR -> RGB
 
 
 detector = load_model()