Update app.py

app.py

@@ -59,6 +59,13 @@ def tokenize_text(
             "cls_token",
             "unk_token",
             "mask_token",
+            "image_token",
+            "audio_token",
+            "video_token",
+            "vision_bos_token",
+            "vision_eos_token",
+            "audio_bos_token",
+            "audio_eos_token",
         ]
         special_tokens = {}
         for token_name in sp_token_list:
@@ -136,24 +143,21 @@ char_count = gr.Number(label="Character Count", value=0, interactive=False)
 with gr.Blocks(title="Token Visualizer", theme="NoCrypt/miku") as webui:
     banner.render()
 
-    with gr.Column():
-        with gr.Row():
-            with gr.Column():
-                model_selector.render()
-                text_input.render()
-                submit_btn.render()
-            with gr.Column():
-                with gr.Accordion("Details", open=False):
-                    with gr.Row():
-                        tokenizer_type.render()
-                        vocab_size.render()
-                    sp_tokens.render()
-                with gr.Row():
-                    token_count.render()
-                    char_count.render()
-
+    with gr.Row(scale=2):
         with gr.Column():
+            model_selector.render()
+            text_input.render()
+            submit_btn.render()
             output_html.render()
+        with gr.Column():
+            with gr.Accordion("Details", open=False):
+                with gr.Row():
+                    tokenizer_type.render()
+                    vocab_size.render()
+                sp_tokens.render()
+            with gr.Row():
+                token_count.render()
+                char_count.render()
 
     # 定义CSS样式
     webui.css = """
@@ -165,16 +169,6 @@ with gr.Blocks(title="Token Visualizer", theme="NoCrypt/miku") as webui:
         font-weight: bold !important;
         color: #2c3e50 !important;
     }
-    .gradio-container { /* 针对 Gradio 的主容器 */
-        width: 100%; /* 根据需要调整宽度 */
-        display: flex;
-        justify-content: center;
-        align-items: center;
-    }
-    .gradio-container > div { /* 直接子元素，通常包含你的内容 */
-        width: 90%; /* 或者你想要的固定宽度 */
-        max-width: 1200px; /* 设置最大宽度 */
-    }
     """
 
     submit_btn.click(