.

app.py

@@ -5,6 +5,7 @@ from datasets import load_dataset
 from transformers import AutoTokenizer
 
 def ReturnTokens(dataset_name, tokenizer_name="openai-community/gpt2", split="train"):
+
     # Initialize tokenizer
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
     
@@ -34,26 +35,21 @@ with gr.Blocks(title="Dataset token counter") as app:
         tokenizer = gr.Textbox(label="Tokenizer", elem_id="tokenizer", info="", placeholder="openai-community/gpt2", value="openai-community/gpt2")
         split = gr.Textbox(label="Split (default: train)", elem_id="split", info="", placeholder="train", value="train")
         tokens = gr.Label(label="Tokens", elem_id="tokens")
-        prompt.submit().success(
-            ReturnTokens,
-            inputs=[prompt,tokenizer,split],
-            outputs=[tokens]
-        )
 
-        gr.on(
-            triggers=[
-                prompt.submit,
-                tokenizer.submit,
-                split.submit,
-            ],
-            fn=ReturnTokens,
-            inputs=[
-                prompt,
-                tokenizer,
-                split
-            ],
-            outputs=[tokens],
-            api_name="run",
-        )
+    gr.on(
+        triggers=[
+            prompt.submit,
+            tokenizer.submit,
+            split.submit,
+        ],
+        fn=ReturnTokens,
+        inputs=[
+            prompt,
+            tokenizer,
+            split
+        ],
+        outputs=[tokens],
+        api_name="run",
+    )
 
 app.launch()