support unsloth templates in URL

app.py

@@ -56,7 +56,14 @@ col.image("https://raw.githubusercontent.com/unslothai/unsloth/main/images/made%
 model = st.text_input("Enter HuggingFace model name", st.query_params.get("model", "Qwen/Qwen2-VL-7B-Instruct"))
 processor = AutoProcessor.from_pretrained(model, trust_remote_code=True)
 text_tokenizer = processor if not hasattr(processor, "tokenizer") else processor.tokenizer
-chat_template_key = st.selectbox("Select chat template", ["model_default"] + list(CHAT_TEMPLATES.keys()), index=0)
+chat_template_predefined = st.query_params.get("chat_template_idx", None)
+possible_templates = ["model_default"] + sorted(CHAT_TEMPLATES.keys())
+if chat_template_predefined is not None:
+    chat_template_idx = possible_templates.index(chat_template_predefined)
+else:
+    chat_template_idx = 0
+
+chat_template_key = st.selectbox("Select chat template", possible_templates, index=chat_template_idx)
 
 if chat_template_key == "model_default":
     chat_template = None
@@ -105,5 +112,5 @@ st.markdown("#### Masked Prompt ('x' is the mask token)")
 st.code(masked_text, language="html")
 
 st.markdown("#### You may share the following URL with others to show them the results")
-url = f"https://zeel-twro.hf.space?model={model}&message={message}&instruction_part={instruction_part}&response_part={response_part}"
+url = f"https://zeel-twro.hf.space?model={model}&message={message}&instruction_part={instruction_part}&response_part={response_part}&chat_template_idx={chat_template_key}"
 st.markdown(f"`{url}`")