Update app.py

app.py

@@ -54,4 +54,20 @@ demo = gr.Interface(
     ]
 )
 
+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.launch()