Update app.py

app.py

@@ -174,7 +174,7 @@ 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], current_bpe[0]
+        return image, current_vis[0], bpe_text[0]
 
     
     run_btn.click(