Update app.py

app.py

@@ -394,7 +394,7 @@ def generate(text_list, image_list, max_new_tokens, force_boi, force_bbox, force
     if args.has_bbox:
         bboxes = extract_box(generated_text)
         if bboxes is not None and len(input_images) > 0:
-            image_viz = visualize_bbox(input_images[0], bboxes)
+            image_viz = visualize_bbox(input_images[-1], bboxes)
             image_base64 = encode_image(image_viz)
             gen_imgs_base64_list.append(image_base64)
             generated_text = re.sub(r'\[\[ <box_start>.*?<box_end>.*?\]\]', 'the green bounding box', generated_text)