Update app.py

app.py

@@ -122,6 +122,13 @@ 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):
+    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:
+        return None, "索引超出范围"
+
 # Gradio界面
 with gr.Blocks(title="BPE Visualization Demo") as demo:
     gr.Markdown("## BPE Visualization Demo - TokenFD基座模型能力可视化")
@@ -202,13 +209,10 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     # )
 
     index_slider.change(
-    lambda x: (
-        print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(current_bpe)}"),  # 打印调试信息
-        (current_vis[x], format_bpe_display(current_bpe[x])) if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe) else (None, "索引超出范围")
-    ),
-    inputs=index_slider,
-    outputs=[heatmap, bpe_display]
-    )
+            update_slider_index,
+            inputs=index_slider,
+            outputs=[heatmap, bpe_display]
+        )
 
 if __name__ == "__main__":
     demo.launch()