Update app.py

app.py

@@ -183,11 +183,14 @@ with gr.Blocks() as demo:
             train_btn = gr.Button("Train Tokenizer")
 
         progress = gr.Textbox(label="Progress", interactive=False, lines=10)
+        decoded_text = gr.Textbox(label="Decoded Text", interactive=False)
+        token_distribution = gr.Image(label="Token Distribution")
 
     start_btn.click(collect_samples, [dataset_name, configs, split, chunk_size], progress)
     stop_btn.click(stop_collection, [], progress)
     restart_btn.click(restart_collection, [], progress)
     analyze_btn.click(analyze_checkpoint, [], progress)
-    train_btn.click(train_tokenizer_fn, [dataset_name, configs, split, vocab_size, min_freq, test_text], [progress])
+    train_btn.click(train_tokenizer_fn, [dataset_name, configs, split, vocab_size, min_freq, test_text], 
+                    [progress, decoded_text, token_distribution])
 
 demo.launch()