applied chat template

app.py

@@ -3,7 +3,10 @@ import streamlit as st
 
 # Access the secret token
 hf_token = st.secrets["default"]["hf_token"]
-st.write(hf_token)
+st.write(f"{hf_token = }")
+
+@st.cache_resource
+def context_text(text): return f"### Context\n{text}\n\n### Answer"
 
 @st.cache_resource
 def load_model():
@@ -13,10 +16,12 @@ model = load_model()
 
 st.header("Test Environment for tiny_CaLL_r10_O1_f10_c1022")
 user_input = st.text_input("Enter your Prompt here:", "")
+contexted_ipnut = context_text(user_input)
+
 
 if user_input:
     with st.spinner('Generating response...'):
-        response = model(user_input, max_length=200, num_return_sequences=1)
+        response = model(contexted_ipnut, max_length=200, num_return_sequences=1)
         generated_text = response[0]['generated_text']
         st.write("Generated Text:")
         st.write(generated_text)