Update task file

src/task.py

@@ -1,4 +1,6 @@
+# Import necessary libraries
 import copy
+import supervision as sv
 from src.utils import clean_text, draw_ocr_bboxes
 from src.model import run_example
 
@@ -26,9 +28,10 @@ def ocr_task(image):
     # Get OCR with region
     ocr_with_region_results = run_example(ocr_with_region_prompt, image)
     output_image = copy.deepcopy(image)
-    output_image = draw_ocr_bboxes(
-        output_image, ocr_with_region_results["<OCR_WITH_REGION>"]
+    detections = sv.Detections.from_lmm(
+        lmm=sv.LMM.FLORENCE_2, result=ocr_with_region_results, resolution_wh=image.size
     )
+    output_image = draw_ocr_bboxes(image, detections)
 
     # Return the output image and cleaned OCR text
     return output_image, cleaned_text