Update app.py

app.py

@@ -33,6 +33,22 @@ def draw_detections(image, detections):
     final_pil_image = Image.fromarray(final_image)
     return final_pil_image
 
+def get_pipeline_prediction(pil_image):
+    try:
+        # Run the object detection pipeline
+        pipeline_output = od_pipe(pil_image)
+        
+        # Draw the detection results on the image
+        processed_image = draw_detections(pil_image, pipeline_output)
+        
+        # Provide both the image and the JSON detection results
+        return processed_image, pipeline_output
+    except Exception as e:
+        # Log the error
+        print(f"An error occurred: {str(e)}")
+        # Return a message and an empty JSON
+        return pil_image, {"error": str(e)}
+
 demo = gr.Interface(
     fn=get_pipeline_prediction,
     inputs=gr.Image(label="Input image", type="pil"),