Update app.py

app.py

@@ -24,7 +24,9 @@ def process_image(pilimg):
     # print(x.shape)
     result = detection_model.predict(source, conf=0.5)
     img_bgr = result[0].plot()
-    return Image.fromarray(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)) 
+    out_pilimg = Image.fromarray(img_bgr[..., ::-1])  # RGB-order PIL image
+    
+    return out_pilimg
 
 def process_video(video):
     # Open the video file
@@ -72,7 +74,7 @@ message = "<h1>Welcome to the Image and Video Upload App For Violin & Guzheng!</
 
 # Create the interface for image upload
 image_interface = gr.Interface(fn=process_image, 
-                               inputs=gr.Image(label="Upload an Image"), 
+                               inputs=gr.Image(label="Upload an Image", type="pil"), 
                                outputs="image")
 
 # Create the interface for video upload