Update app.py

app.py

@@ -35,6 +35,22 @@ def draw_bounding_boxes(image, bounding_boxes, outline_color="red", line_width=2
     return image
 
 
+def rescale_bounding_boxes(bounding_boxes, original_width, original_height, scaled_width=1000, scaled_height=1000):
+    x_scale = original_width / scaled_width
+    y_scale = original_height / scaled_height
+    rescaled_boxes = []
+    for box in bounding_boxes:
+        xmin, xmax, ymin, ymax = box
+        rescaled_box = [
+            xmin * x_scale,
+            xmax * x_scale,
+            ymin * y_scale,
+            ymax * y_scale
+        ]
+        rescaled_boxes.append(rescaled_box)
+    return rescaled_boxes
+
+
 @spaces.GPU
 def run_example(image, text_input, model_id="Qwen/Qwen2-VL-7B-Instruct"):
     model = models[model_id].eval()
@@ -75,7 +91,8 @@ def run_example(image, text_input, model_id="Qwen/Qwen2-VL-7B-Instruct"):
     pattern = r'\[\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\]'
     matches = re.findall(pattern, str(output_text))
     parsed_boxes = [[int(num) for num in match] for match in matches]
-    return output_text, parsed_boxes, draw_bounding_boxes(image, parsed_boxes)
+    scaled_boxes = rescale_bounding_boxes(parsed_boxes)
+    return output_text, parsed_boxes, draw_bounding_boxes(image, scaled_boxes)
 
 css = """
   #output {