Update app.py

app.py

@@ -400,7 +400,7 @@ def generate(text_list, image_list, max_new_tokens, force_boi, force_bbox, force
             if '<box_start>' in generated_text:
                 generated_text = re.sub(r'\[\[ <box_start>.*?<box_end>.*?\]\]', 'the green bounding box', generated_text)
             else:
-                generated_text = re.sub(r'\[\[.*?<box_end>.*?\]\]', 'the green bounding box', generated_text)
+                generated_text = re.sub(r'<loc-\d+> <loc-\d+> <loc-\d+> <loc-\d+> <box_end>  \]\]', 'the green bounding box', generated_text)
             generated_text += IMG_FLAG
     print(input_text + generated_text)
     return {'text': generated_text, 'images': gen_imgs_base64_list, 'error_msg': error_msg}