Update app.py

app.py

@@ -174,13 +174,13 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
         print("current_vis",len(current_vis))
         print("current_bpe",len(current_bpe))
         
-        return image, current_vis[0], bpe_text[0], gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+        return image, current_vis[0],f"Current BPE: {current_bpe[0]}", gr.update(visible=True), gr.update(visible=True)
 
     
     run_btn.click(
         on_run_clicked,
         inputs=[model_type, image_input, text_input],
-        outputs=[orig_img, heatmap, bpe_display, prev_btn, next_btn, bpe_display],  # 让它们显示
+        outputs=[orig_img, heatmap, bpe_display, prev_btn, next_btn],  # 让它们显示
     )
     
     prev_btn.click(