Update app.py

app.py

@@ -11,6 +11,10 @@ processor = DetrImageProcessor.from_pretrained('facebook/detr-resnet-101')
 model = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-101')
 
 def object_detection(image, confidence_threshold):
+    # Convert the input to a PIL Image object if it's not already
+    if not isinstance(image, Image.Image):
+        image = Image.open(io.BytesIO(image))
+    
     # Preprocess the image
     inputs = processor(images=image, return_tensors="pt")
 
@@ -56,7 +60,7 @@ def object_detection(image, confidence_threshold):
 # Define the Gradio interface
 demo = gr.Interface(
     fn=object_detection,
-    inputs=[gr.Image(label="Upload an Image"), gr.Slider(minimum=0.0, maximum=1.0, label="Confidence Threshold")],
+    inputs=[gr.Image(label="Upload an Image"), gr.Slider(minimum=0.0, maximum=1.0, label="Confidence Threshold", default=0.5)],
     outputs=[gr.Image(label="Detected Objects"), gr.Textbox(label="Detected Objects List")],
     title="Object Detection with DETR (ResNet-101)",
     description="Upload an image and get object detection results using the DETR model with a ResNet-101 backbone."