Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 import base64
 from PIL import Image
 from io import BytesIO
+import re
 
 
 models = {
@@ -62,7 +63,11 @@ def run_example(image, text_input, model_id="Qwen/Qwen2-VL-7B-Instruct"):
     output_text = processor.batch_decode(
         generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
     )
-    return output_text
+
+    pattern = r'\[\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\]'
+    matches = re.findall(pattern, string)
+    result = [[int(num) for num in match] for match in matches]
+    return result
 
 css = """
   #output {