Update app.py

app.py

@@ -55,10 +55,12 @@ def generate_tokens(text):
     input_ids = input["input_ids"].tolist()[0]
     output_ids = output.tolist()[0]
 
+    formatted_output = [format(x, 'd') for x in output_ids]
+
     input_tokens_str = tokenizer.convert_ids_to_tokens(input_ids)
     #output_tokens_str = tokenizer.convert_tokens_to_ids(output_ids)
 
-    return " ".join(input_tokens_str), " ".join(output_ids)
+    return " ".join(input_tokens_str), " ".join(formatted_output)
 
 """
 For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface