Update app.py

app.py

@@ -1,13 +1,25 @@
 import gradio as gr
+from transformers import AutoTokenizer
 
 with gr.Blocks() as demo:
-    huggingface = gr.Textbox(placeholder="mistralai/Mistral-Small-24B-Instruct-2501", show_label=True, label="huggingface ID")
+    huggingface = gr.Textbox(value="mistralai/Mistral-Small-24B-Instruct-2501", show_label=True, label="huggingface ID")
     button = gr.Button(value="get chat template")
-    textbox = gr.Textbox(lines=10.0)
+    output = gr.Textbox(lines=10.0, show_copy_button=True, visible=False)
 
-    @button.click(inputs=[button], outputs=[])
-    def fn_1(button):
-        ...
-        return 
+    @button.click(inputs=[huggingface], outputs=[output])
+    def submit(huggingface_id):
+        try:
+            template = AutoTokenizer.from_pretrained(huggingface_id).apply_chat_template(
+                [
+                    {"role": "system", "content": "system-prompt"},
+                    {"role": "user", "content": "user-prompt"},
+                    {"role": "assistant", "content": "assistant-prompt"}
+                ],
+                tokenize=False
+            )
+            return gr.update(value=template, visible=True)
+        except:
+            raise gr.Error("Could not get chat template")
+        return gr.update(visible=False)
 
 demo.launch()