add example

AI-Yash · Dec 28, 2021 · d7a4e6e · d7a4e6e
1 parent 13e8e54
commit d7a4e6e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -1,6 +1,7 @@
 # st-chat
 
-Streamlit Component, for a Chat bot UI
+Streamlit Component, for a Chat-bot UI, [example app](https://share.streamlit.io/ai-yash/st-chat/main/examples/chatbot.py)
+
 authors - [@yashppawar](https://github.com/yashppawar) & [@YashVardhan-AI](https://github.com/yashvardhan-ai)
 
 ## Installation

diff --git a/examples/chatbot.py b/examples/chatbot.py
@@ -2,9 +2,17 @@
 from streamlit_chat import message
 import requests
 
+st.set_page_config(
+    page_title="Streamlit Chat - Demo",
+    page_icon=":robot:"
+)
+
 API_URL = "https://api-inference.huggingface.co/models/facebook/blenderbot-400M-distill"
 headers = {"Authorization": st.secrets['api_key']}
 
+st.header("Streamlit Chat - Demo")
+st.markdown("[Github](https://github.com/ai-yash/st-chat")
+
 if 'generated' not in st.session_state:
     st.session_state['generated'] = []