Update app.py

app.py

@@ -122,8 +122,10 @@ def format_bpe_display(bpe):
     # 使用HTML标签来设置字体大小、颜色，加粗，并居中
     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: {current_vis}, current_bpe length: {current_bpe}")
+@spaces.GPU
+def update_slider_index(x):
+    global current_vis, current_bpe, current_index
+    print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(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 +209,10 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     inputs=index_slider,
     #     outputs=[heatmap, bpe_display]
     # )
-    vis = gr.State(value=current_vis)  # 初始化状态
-    bpe = gr.State(value=current_bpe)  # 初始化状态
-    
+
     index_slider.change(
             update_slider_index,
-            inputs=[index_slider,vis,bpe],
+            inputs=index_slider,
             outputs=[heatmap, bpe_display]
         )