Let's try partial()

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer
 import json
+from functools import partial
 
 tokenizer = AutoTokenizer.from_pretrained("HuggingFaceH4/zephyr-7b-beta")
 
@@ -146,7 +147,7 @@ with gr.Blocks() as demo:
                 inputs=[template_in, conversation_in, generation_prompt_check, cleanup_whitespace_check],
                 outputs=[formatted_out, code_snippet_out]
                 )
-    load_chatml.click(fn=load_template, inputs=["chatml"])
+    load_chatml.click(fn=partial(load_template, "chatml"))
     load_zephyr.click(fn=load_template, inputs=["zephyr"])
     load_llama.click(fn=load_template, inputs=["llama"])
     load_alpaca.click(fn=load_template, inputs=["alpaca"])