Update app.py

app.py

@@ -1,4 +1,5 @@
-from transformers import DetrFeatureExtractor, DetrForObjectDetection
+from transformers import DetrFeatureExtractor, DetrForObjectDetection, 
+from transformers import pipeline
 from PIL import Image, ImageDraw, ImageFont
 import requests
 import gradio as gr
@@ -9,7 +10,7 @@ def draw_bounding_box(im, score, label, xmin, ymin, xmax, ymax, index, num_boxes
   # Draw the actual bounding box
   outline = 'blue'
   im_with_rectangle = ImageDraw.Draw(im)  
-  im_with_rectangle.rounded_rectangle((xmin, ymin, xmax, ymax), outline = outline, width = 3, radius = 10)
+  im_with_rectangle.rounded_rectangle((xmin, ymin, xmax, ymax), outline = outline, width = 2, radius = 7)
 
   # Return the result
   return im
@@ -35,7 +36,7 @@ def object_classify(img):
                                   box['xmin'], box['ymin'],
                                   box['xmax'], box['ymax'],
                                   index, num_boxes)
-  index += 1
+    index += 1
   return output_image