maxiw commited on
Commit
f3a13dd
·
1 Parent(s): de6a4a2

correct text

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -85,6 +85,7 @@ def run_example(image, text_input, system_prompt, model_id="OS-Copilot/OS-Atlas-
85
  generated_ids_trimmed, skip_special_tokens=False, clean_up_tokenization_spaces=False
86
  )
87
  print(output_text)
 
88
 
89
  object_ref_pattern = r"<\|object_ref_start\|>(.*?)<\|object_ref_end\|>"
90
  box_pattern = r"<\|box_start\|>(.*?)<\|box_end\|>"
 
85
  generated_ids_trimmed, skip_special_tokens=False, clean_up_tokenization_spaces=False
86
  )
87
  print(output_text)
88
+ text = output_text[0]
89
 
90
  object_ref_pattern = r"<\|object_ref_start\|>(.*?)<\|object_ref_end\|>"
91
  box_pattern = r"<\|box_start\|>(.*?)<\|box_end\|>"