Update app.py

app.py

@@ -1,4 +1,3 @@
-import time
 import json 
 import os
 
@@ -31,8 +30,24 @@ def generate(prompt):
         yield "<<Some errors occured>>"
 
 
+@st.experimental_dialog("Disclaimer")
+def disclaimer():
+    st.write("The response of our LLM may be incorrect, harmful, or biased.")
+    st.write("We may use users' chat data in this site to improve our LLM.")
+
+
 st.title("Stockmark-LLM-100b")
 
+intro = """This is a demo site for Stockmark-LLM-100b.
+- Pretrained model: [stockmark/stockmark-100b](https://huggingface.co/stockmark/stockmark-100b)
+- Instruction tuned model: [stockmark/stockmark-100b-instruct-v0](https://huggingface.co/stockmark/stockmark-100b-instruct-v0.1)
+"""
+
+st.markdown(intro)
+
+if st.button("Disclaimer"):
+    disclaimer()
+
 prompt = st.session_state.get("prompt", "")
 response = st.session_state.get("response", "")