Update app.py

app.py

@@ -56,9 +56,9 @@ def generate_tokens(text):
     output_ids = output.tolist()[0]
 
     input_tokens_str = tokenizer.convert_ids_to_tokens(input_ids)
-    output_tokens_str = output_ids
+    #output_tokens_str = tokenizer.convert_tokens_to_ids(output_ids)
 
-    return " ".join(input_tokens_str), " ".join(output_tokens_str)
+    return " ".join(input_tokens_str), " ".join(output_ids)
 
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
@@ -101,15 +101,15 @@ with gr.Blocks() as demo:
     with gr.Row():
         input_text = gr.Textbox(label="Input text")
         input_tokens = gr.Textbox(label="Input tokens")
-        output_tokens = gr.Textbox(label="Output tokens")
+        output_ids = gr.Textbox(label="Output tokens")
 
         def update_tokens(input_text):
-            input_tokens_str, output_tokens_str = generate_tokens(input_text)
-            return input_tokens_str, output_tokens_str
+            input_tokens_str, output_ids = generate_tokens(input_text)
+            return input_tokens_str, output_ids
 
         input_text.change(update_tokens, 
                           inputs=input_text,
-                          outputs=[input_tokens, output_tokens])
+                          outputs=[input_tokens, output_ids])
 
 if __name__ == "__main__":
     demo.launch(debug=True)