Update app.py

app.py

@@ -1,25 +1,26 @@
 import gradio as gr
 from PIL import Image
-import cv2
 import numpy as np
 from ultralytics import YOLO
-import io
 
 def load_model():
-    model = YOLO('yolov8x.pt')
-    return model
+    return YOLO('yolov8x.pt')
 
 model = load_model()
 
 def process_image(input_image):
-    open_cv_image = np.array(input_image)
-    open_cv_image = open_cv_image[:, :, ::-1]  # Convert RGB to BGR
-    results = model(open_cv_image)
+    # Convert PIL Image to numpy array
+    np_image = np.array(input_image)
+    
+    # Run YOLO model
+    results = model(np_image)
+    
     # Plot results on image
     plotted_img = results[0].plot()
-    # Convert BGR to RGB
-    rgb_image = cv2.cvtColor(plotted_img, cv2.COLOR_BGR2RGB)
-    pil_image = Image.fromarray(rgb_image)
+    
+    # Convert numpy array back to PIL Image
+    pil_image = Image.fromarray(plotted_img)
+    
     return pil_image
 
 iface = gr.Interface(
@@ -29,4 +30,6 @@ iface = gr.Interface(
     title="Object Detection with YOLO",
     description="Upload an image and the model will detect objects."
 )
-iface.launch(share=False)
+
+if __name__ == "__main__":
+    iface.launch(share=False)