Update app.py

app.py

@@ -40,6 +40,9 @@ def draw_detections(image, detections):
     return final_pil_image
 
 def get_pipeline_prediction(pil_image):
+    # Ensure the image is a PIL Image as expected by the model pipeline
+    if not isinstance(pil_image, Image.Image):
+        pil_image = Image.fromarray(pil_image.astype('uint8'), 'RGB')
     try:
         pipeline_output = od_pipe(pil_image)
         processed_image = draw_detections(pil_image, pipeline_output)
@@ -52,7 +55,7 @@ def get_pipeline_prediction(pil_image):
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            inp_image = gr.Image(label="Input image")  # Removed the 'tool=None' part
+            inp_image = gr.Image(label="Input image", type="pil")  # Ensure the input is loaded as a PIL Image
             btn_run = gr.Button('Run Detection')
         with gr.Column():
             with gr.Tab("Annotated Image"):