Update app.py

app.py

@@ -29,37 +29,54 @@ def generate(prompt):
     except:
         yield "<<Some errors occured>>"
 
-
-st.title("Stockmark-LLM-100b")
-
 intro = """This is a demo site for Stockmark-LLM-100b. This service is running on AWS Inferentia2.
 - Pretrained model: [stockmark/stockmark-100b](https://huggingface.co/stockmark/stockmark-100b)
 - Instruction tuned model: [stockmark/stockmark-100b-instruct-v0.1](https://huggingface.co/stockmark/stockmark-100b-instruct-v0.1)
 """
 
-st.markdown(intro)
+disclaimer = """
+- Responses of our LLM may be incorrect, biased, or harmful.
+- We may use users chat data in this demo to improve our LLM.
+"""
+
+
+tab1, tab2, tab3 = st.tabs(["Demo", "Disclaimer"])
 
-prompt = st.session_state.get("prompt", "")
-response = st.session_state.get("response", "")
 
-if prompt == "" or response:
-    print("new_session")
-    prompt_new = st.text_area("Prompt:")
-    if prompt_new:
-        st.session_state["prompt"] = prompt_new
-        st.session_state["response"] = ""
-        st.rerun()
-else:
-    prompt = st.text_area("Prompt:", value=prompt, disabled=True)
 
-if prompt:
 
-    if response:
-        with st.chat_message("assistant"):
-            st.write(response)
+with tab1:
+    st.title("Stockmark-LLM-100b")
+    
+    
+    
+    st.markdown(intro)
+    
+    prompt = st.session_state.get("prompt", "")
+    response = st.session_state.get("response", "")
+    
+    if prompt == "" or response:
+        print("new_session")
+        prompt_new = st.text_area("Prompt:")
+        if prompt_new:
+            st.session_state["prompt"] = prompt_new
+            st.session_state["response"] = ""
+            st.rerun()
     else:
-        with st.chat_message("assistant"):
-            response = st.write_stream(generate(prompt))
-        
-        st.session_state["response"] = response
-        st.rerun()
+        prompt = st.text_area("Prompt:", value=prompt, disabled=True)
+    
+    if prompt:
+    
+        if response:
+            with st.chat_message("assistant"):
+                st.write(response)
+        else:
+            with st.chat_message("assistant"):
+                response = st.write_stream(generate(prompt))
+            
+            st.session_state["response"] = response
+            st.rerun()
+
+with tab2:
+    st.title("Stockmark-LLM-100b: Disclaimer")
+    st.markdown(disclaimer)