Update app.py

app.py

@@ -8,14 +8,18 @@ tokenizer = LlamaTokenizer.from_pretrained("llama_tokenizer")
 def tokenize(prompt):
     tokens = tokenizer.encode(prompt, add_special_tokens=False)
     num_tokens = len(tokens)
-    return ("Number of tokens:", num_tokens)
+    return num_tokens
 
 with gr.Blocks() as demo:
-    gr.Markdown("Token Counter for LLAMA")
+    gr.Markdown(
+    """
+    # Token Counter for LLAMA
+    """)
     with gr.Row():
         inp = gr.Textbox(placeholder="Enter prompt")
-        out = gr.Textbox()
-    btn = gr.Button("Run")
+        with gr.Column():
+            out = gr.Textbox(label="Number of tokens")
+            btn = gr.Button("Run")
     btn.click(fn=tokenize, inputs=inp, outputs=out)
 
 demo.launch()