Update app.py

app.py

@@ -19,8 +19,9 @@ model = YOLO("best.pt")  # load a custom model
 def gradio_wrapper(img):
     global model
     #print(np.shape(img))
-    results = model(img,show = True)  # predict on an image
-    return results[0]
+    results = model.predict(img)  # predict on an image
+    
+    return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 demo = gr.Interface(
     gradio_wrapper,
     #gr.Image(source="webcam", streaming=True, flip=True),