Update app.py

app.py

@@ -79,6 +79,7 @@ if prompt := st.chat_input("Type your message..."):
 
             if output and isinstance(output, list) and 'generated_text' in output[0]:
                 assistant_response = output[0]['generated_text'].strip()
+                assistant_response = assistant_response.replace("</think>", "").strip()
                 st.markdown(f"**🤖 AI:** {assistant_response}")
                 st.session_state.messages.append({"role": "assistant", "content": assistant_response})
             else: