Add PR output

app.py

@@ -28,7 +28,14 @@ def apply_chat_template(template, test_conversation, add_generation_prompt, clea
     tokenizer.chat_template = template
     outputs = []
     conversation = json.loads(test_conversation)
-    return tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=add_generation_prompt)
+    pr_snippet = (
+        "tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT)",
+        f"tokenizer.chat_template = \"{template}\"",
+        tokenizer.push_to_hub(CHECKPOINT, create_pr=True)
+    )
+    pr_snippet = "\n".join(pr_snippet)
+    formatted = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=add_generation_prompt)
+    return formatted, pr_snippet
 
 iface = gr.Interface(
     description=description_text,
@@ -39,5 +46,8 @@ iface = gr.Interface(
         gr.Checkbox(value=False, label="Add generation prompt"),
         gr.Checkbox(value=True, label="Cleanup template whitespace"),
     ],
-    outputs=gr.TextArea(label="Formatted conversation"))
+    outputs=[
+        gr.TextArea(label="Formatted conversation")),
+        gr.TextArea(label="Code snippet to create PR", lines=3, show_copy_button=True))
+    ]
 iface.launch()