message

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from streamlit_chat import message
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
 model_dir = "lcw99/t5-base-korean-chit-chat"
@@ -7,6 +8,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_dir)
 
 st.title("Summarize Text")
 chat_hist = st.empty()
+message('안녕?')
 chat_hist.text_area("Chat history:", "")
 sentence = st.text_area('Please enter your message :', height=50)
 button = st.button("Send")
@@ -42,4 +44,5 @@ with st.spinner("Generating chat.."):
           bot_text = bot_text.replace("\n", " / ")
           chat_history.append("B: " + bot_text)
           
-          chat_hist.text_area("Chat history:", hist)
+          chat_hist.text_area("Chat history:", hist)
+          message(bot_text)

requirements.txt

@@ -1,2 +1,3 @@
 transformers
-torch
+torch
+streamlit-chat