feat: set template

src/streamlit_app.py

@@ -74,6 +74,11 @@ def main():
             user_prompt = alpaca_template.format(query, "")
             if AI_MODE == "ON":
                 # for chat models
+                tokenizer.chat_template = {
+                    "role": "user",
+                    "prompt": user_prompt,
+                    "generation_prompt": "",
+                }
                 user_prompt = tokenizer.apply_chat_template(
                     user_prompt, tokenize=False, add_generation_prompt=True)
                 outputs = processor(user_prompt,