add Annotator

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 # load libraries
 from huggingface_hub import hf_hub_download
 from ultralytics import YOLO
-from supervision import Detections
+import supervision as sv
 from PIL import Image
 
 # download model
@@ -19,13 +19,15 @@ import gradio as gr
 
 def sepia(input_img):
     output = model(input_img)
-    results = Detections.from_ultralytics(output[0])
-    arr_int = results.xyxy.astype(int)
+    results = sv.Detections.from_ultralytics(output[0])
 
-    for x, y, x2, y2 in arr_int:
-      cv2.rectangle(input_img, (x, y), (x2, y2), (0, 255, 0), 2)
+    bounding_box_annotator = sv.BoundingBoxAnnotator()
+    annotated_frame = bounding_box_annotator.annotate(
+        scene = input_img.copy(),
+        detections = results
+    )
 
-    return input_img
+    return annotated_frame
 
 demo = gr.Interface(sepia, gr.Image(), "image")
 if __name__ == "__main__":