Update app.py

app.py

@@ -3,17 +3,13 @@ from transformers import pipeline
 from PIL import Image, ImageDraw, ImageFont
 import tempfile
 
-# Load the YOLOS object detection model
 detector = pipeline("object-detection", model="hustvl/yolos-small")
 
-# Define some colors to differentiate classes
 COLORS = ["red", "blue", "green", "orange", "purple", "yellow", "cyan", "magenta"]
 
-# Helper function to assign color per label
 def get_color_for_label(label):
     return COLORS[hash(label) % len(COLORS)]
 
-# Main function: detect, draw, and return outputs
 def detect_and_draw(image, threshold):
     results = detector(image)
     image = image.convert("RGB")
@@ -46,14 +42,11 @@ def detect_and_draw(image, threshold):
         box_coords = (box["xmin"], box["ymin"], box["xmax"], box["ymax"])
         annotations.append((box_coords, label))
 
-    # Save image for download
     temp_file = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
     image.save(temp_file.name)
 
-    # ✅ Return the (image, annotations) tuple and the path to the saved image
+    
     return (image, annotations), temp_file.name
-
-# Gradio UI setup
 demo = gr.Interface(
     fn=detect_and_draw,
     inputs=[