Update app.py

app.py

@@ -14,45 +14,39 @@ def get_color_for_label(label):
 
 # Main function: detect, draw, and return outputs
 def detect_and_draw(image, threshold):
-    try:
-        # Perform object detection
-        results = detector(image)
-        image = image.convert("RGB")
-        draw = ImageDraw.Draw(image)
-
-        # Try to load a font for annotations, else use default
-        try:
-            font = ImageFont.truetype("arial.ttf", 16)
-        except:
-            font = ImageFont.load_default()
+    results = detector(image)
+    image = image.convert("RGB")
+    draw = ImageDraw.Draw(image)
 
-        annotations = []
+    try:
+        font = ImageFont.truetype("arial.ttf", 16)
+    except:
+        font = ImageFont.load_default()
 
-        for obj in results:
-            score = obj["score"]
-            if score < threshold:
-                continue
+    annotations = []
 
-            label = f"{obj['label']} ({score:.2f})"
-            box = obj["box"]
-            color = get_color_for_label(obj["label"])
+    for obj in results:
+        score = obj["score"]
+        if score < threshold:
+            continue
 
-            # Draw the bounding box and label
-            draw.rectangle(
-                [(box["xmin"], box["ymin"]), (box["xmax"], box["ymax"])],
-                outline=color,
-                width=3,
-            )
+        label = f"{obj['label']} ({score:.2f})"
+        box = obj["box"]
+        color = get_color_for_label(obj["label"])
 
-            draw.text((box["xmin"] + 5, box["ymin"] + 5), label, fill=color, font=font)
+        draw.rectangle(
+            [(box["xmin"], box["ymin"]), (box["xmax"], box["ymax"])],
+            outline=color,
+            width=3,
+        )
 
-            box_coords = (box["xmin"], box["ymin"], box["xmax"], box["ymax"])
-            annotations.append((box_coords, label))
+        draw.text((box["xmin"] + 5, box["ymin"] + 5), label, fill=color, font=font)
 
-        return image, annotations
+        box_coords = (box["xmin"], box["ymin"], box["xmax"], box["ymax"])
+        annotations.append((box_coords, label))
 
-    except Exception as e:
-        return f"Error during detection: {e}", None
+    # Return the annotated image and annotations (no download option)
+    return image, annotations
 
 # Gradio UI setup
 demo = gr.Interface(
@@ -66,7 +60,6 @@ demo = gr.Interface(
     ],
     title="YOLOS Object Detection",
     description="Upload an image to detect objects using the YOLOS-small model. Adjust the confidence threshold using the slider.",
-    live=True
 )
 
-demo.launch()
+demo.launch()