Update app.py

app.py

@@ -56,7 +56,7 @@ def generate_tokens(text):
     output_ids = output.tolist()[0]
 
     input_tokens_str = tokenizer.convert_ids_to_tokens(input_ids)
-    output_tokens_str = tokenizer.convert_ids_to_tokens(output_ids)
+    output_tokens_str = output_ids
 
     return " ".join(input_tokens_str), " ".join(output_tokens_str)
 
@@ -81,7 +81,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 #)
 
 with gr.Blocks() as demo:
-    with gr.Row():
+    with gr.Column():
         gr.ChatInterface(
             respond,
             additional_inputs=[
@@ -98,7 +98,7 @@ with gr.Blocks() as demo:
             ],
         )
         
-    with gr.Column():
+    with gr.Row():
         input_text = gr.Textbox(label="Input text")
         input_tokens = gr.Textbox(label="Input tokens")
         output_tokens = gr.Textbox(label="Output tokens")