Update app.py

app.py

@@ -8,8 +8,10 @@ def get_response(user_input):
     except Exception as e:
         return "Error: {}".format(str(e))
     else:
-        st.chat_message("You", user_input, avatar=None)
-        st.chat_message("KviGPT", response, avatar=None)
+        with st.chat_message("User"):
+            st.write(user_input)
+        with st.chat_message("KviGPT"):
+            st.write(response)
 
 st.sidebar.title("KviGPT")
 st.sidebar.subheader("Chat based on Gpt-3")