From 474d5c465be97fd136ce598c743bcdbde28e5e39 Mon Sep 17 00:00:00 2001 From: Abdul Samad Date: Tue, 20 Aug 2024 19:52:28 +0500 Subject: [PATCH] Update app.py --- src/app.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/app.py b/src/app.py index 6a43c70..52e5359 100644 --- a/src/app.py +++ b/src/app.py @@ -125,7 +125,11 @@ def main(): with st.chat_message("assistant"): with st.spinner("Generating..."): response = llm_chain.invoke({"input": prompt}) - st.markdown(response) + index = response.find("```") + if index != -1: + st.markdown(response[index:]) + else: + st.markdown(response) message = { "role": "assistant", "content": response,