Update convo.py

convo.py

@@ -4,7 +4,7 @@ from streamlit_option_menu import option_menu
 
 
 def Convo():
-    convo = st.text_input("Enter your message")
+    convo = st.chat_input("Enter your message")
     chatbot = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
     conversation = Conversation(convo)
 
@@ -18,8 +18,9 @@ def Convo():
 
     # conversation.add_user_input("{}".format(convo))
     # conversation = chatbot(conversation)
-    # conversation.generated_responses[-1]
-    st.write(ans)
+    with st.chat_message("assistant"):
+        ans.generated_responses[-1]
+    # st.write(ans)
 
 
 # convo()