Update app.py

app.py

@@ -59,13 +59,8 @@ def upload_image_with_labels(file):
     labels = detect_labels(image)
     return ", ".join(labels)
 
-def switch_interface(file):
-    image = Image.open(file.name)
-    image_with_boxes = detect_objects(image)
-    return image_with_boxes
-
 iface_switch = gr.Interface(
-    fn=switch_interface,
+    fn=upload_image_with_boxes,
     inputs="file",
     outputs="image",
     title="Object Detection with Boxes",
@@ -73,16 +68,12 @@ iface_switch = gr.Interface(
     allow_flagging=False
 )
 
-def show_labels(image):
-    labels = detect_labels(image)
-    return ", ".join(labels)
-
 iface_labels = gr.Interface(
-    fn=show_labels,
-    inputs="image",
+    fn=upload_image_with_labels,
+    inputs="file",
     outputs="text",
     title="Detected Object Labels",
-    description="Tap the image to switch back to object detection with boxes view.",
+    description="Upload an image and get the detected object labels using DETR model. Tap the text to switch back to object detection with boxes view.",
     allow_flagging=False
 )