Update app.py

app.py

@@ -35,11 +35,30 @@ model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", config
 image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
 od_pipe = pipeline(task='object-detection', model=model, image_processor=image_processor)
 
+##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)}
 def get_pipeline_prediction(pil_image):
     try:
         # Run the object detection pipeline
         pipeline_output = od_pipe(pil_image)
-        
+
+        # Debugging: print the keys in the output dictionary
+        if pipeline_output:
+            print("Keys available in the detection output:", pipeline_output[0].keys())
+
         # Draw the detection results on the image
         processed_image = draw_detections(pil_image, pipeline_output)
         
@@ -50,7 +69,7 @@ def get_pipeline_prediction(pil_image):
         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"),