Update app.py

app.py

@@ -34,7 +34,9 @@ def model_answer(user_input):
         'user_input': user_input
     })
     result = model.invoke(prompt)
-    return result.content.strip()
+    match = re.search(r"<\|assistant\|>(.*)", result.content, re.DOTALL)
+    ai_resp = match.group(1).strip()
+    return ai_resp
 
 prompt = st.chat_input("Say something")