Update app.py

app.py

@@ -123,7 +123,7 @@ def format_bpe_display(bpe):
     return f"<div style='text-align:center; font-size:20px;'><strong>Current BPE: <span style='color:red;'>{bpe}</span></strong></div>"
 
 def update_slider_index(x, current_vis, current_bpe):
-    print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(current_bpe)}")
+    print(f"x: {x}, current_vis length: {current_vis}, current_bpe length: {current_bpe}")
     if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe):
         return current_vis[x], format_bpe_display(current_bpe[x])
     else:
@@ -207,12 +207,12 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     inputs=index_slider,
     #     outputs=[heatmap, bpe_display]
     # )
-    current_vis = gr.State(value=3)  # 初始化状态
-    current_bpe = gr.State(value=['abc', 'eee', 'asx'])  # 初始化状态
+    vis = gr.State(value=current_vis)  # 初始化状态
+    bpe = gr.State(value=current_bpe)  # 初始化状态
     
     index_slider.change(
             update_slider_index,
-            inputs=[index_slider,current_vis,current_bpe],
+            inputs=[index_slider,vis,bpe],
             outputs=[heatmap, bpe_display]
         )