Update app.py

app.py

@@ -54,18 +54,27 @@ def object_detection(image, confidence_threshold):
     return result_image, detected_objects_text
 
 # Define the Gradio interface
+# demo = gr.Interface(
+#     object_detection,
+#     [
+#         gr.inputs.Image(label="Upload an Image"),
+#         gr.outputs.Image(label="Detected Objects"),
+#         gr.outputs.Textbox(label="Detected Objects List"),
+#         gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Confidence Threshold")
+#     ],
+#     title="Object Detection with DETR (ResNet-101)",
+#     description="Upload an image and adjust the confidence threshold to view detected objects."
+# )
+
 demo = gr.Interface(
-    object_detection,
-    [
-        gr.inputs.Image(label="Upload an Image"),
-        gr.outputs.Image(label="Detected Objects"),
-        gr.outputs.Textbox(label="Detected Objects List"),
-        gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Confidence Threshold")
-    ],
+    fn=object_detection,
+    inputs=[gr.Image(label="Upload an Image"),gr.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Confidence Threshold")],
+    outputs=[gr.Image(label="Detected Objects"),gr.Textbox(label="Detected Objects List")],
     title="Object Detection with DETR (ResNet-101)",
-    description="Upload an image and adjust the confidence threshold to view detected objects."
+    description="Upload an image and get object detection results using the DETR model with a ResNet-101 backbone.",
 )
 
+
 # Launch the Gradio interface
 if __name__ == "__main__":
     demo.launch()