maxiw commited on
Commit
3b6d536
1 Parent(s): 59c0321

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -65,9 +65,9 @@ def run_example(image, text_input, model_id="Qwen/Qwen2-VL-7B-Instruct"):
65
  )
66
 
67
  pattern = r'\[\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\]'
68
- matches = re.findall(pattern, output_text)
69
- result = [[int(num) for num in match] for match in matches]
70
- return result
71
 
72
  css = """
73
  #output {
@@ -87,8 +87,9 @@ with gr.Blocks(css=css) as demo:
87
  text_input = gr.Textbox(label="Description of Localization Target")
88
  submit_btn = gr.Button(value="Submit")
89
  with gr.Column():
90
- output_text = gr.Textbox(label="Output Text")
 
91
 
92
- submit_btn.click(run_example, [input_img, text_input, model_selector], [output_text])
93
 
94
  demo.launch(debug=True)
 
65
  )
66
 
67
  pattern = r'\[\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\]'
68
+ matches = re.findall(pattern, str(output_text))
69
+ parsed_boxes = [[int(num) for num in match] for match in matches]
70
+ return output_text, parsed_boxes
71
 
72
  css = """
73
  #output {
 
87
  text_input = gr.Textbox(label="Description of Localization Target")
88
  submit_btn = gr.Button(value="Submit")
89
  with gr.Column():
90
+ model_output_text = gr.Textbox(label="Model Output Text")
91
+ parsed_boxes = gr.Textbox(label="Parsed Boxes")
92
 
93
+ submit_btn.click(run_example, [input_img, text_input, model_selector], [model_output_text, parsed_boxes])
94
 
95
  demo.launch(debug=True)