Update app.py

app.py

@@ -72,13 +72,11 @@ def response_generator():
     query = f"{prompt}"
     if query != "None":
         for m in chain.stream(query):
-            print(m)
-            yield m + " "
+            yield m
+            time.sleep(0.05)
     else:
         yield "How can i help you?"
         
-            # time.sleep(0.05)
-    # return chain.invoke(query)
 ###########################################
 # Display assistant response in chat message container
 with st.chat_message("assistant"):