Update app.py

app.py

@@ -1,9 +1,9 @@
 from transformers import DetrImageProcessor, DetrForObjectDetection
 import torch
-from PIL import Image, ImageDraw
+from PIL import Image
 import gradio as gr
-import requests
-import random
+# import requests
+# import random
 
 def detect_objects(image):
     # Load the pre-trained DETR model
@@ -18,13 +18,6 @@ def detect_objects(image):
     target_sizes = torch.tensor([image.size[::-1]])
     results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]
 
-    # Draw bounding boxes and labels on the image
-    #draw = ImageDraw.Draw(image)
-    #for i, (score, label, box) in enumerate(zip(results["scores"], results["labels"], results["boxes"])):
-    #    box = [round(i, 2) for i in box.tolist()]
-    #    color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
-    #    draw.rectangle(box, outline=color, width=3)
-    #    draw.text((box[0], box[1]), model.config.id2label[label.item()], fill=color)
     res = []
     for label in results["labels"]:
         res.append(model.config.id2label[label.item()])