lhzstar commited on
Commit
e916883
1 Parent(s): fb6ade2

new commits

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -14,7 +14,11 @@ def main():
14
  st.sidebar.header("CelebChat")
15
  expander = st.sidebar.expander('About the app')
16
  with expander:
17
- st.markdown("This app is a demo of celebrity chatting!")
 
 
 
 
18
  if "messages" not in st.session_state:
19
  st.session_state["messages"] = []
20
  if "QA_model_path" not in st.session_state:
@@ -36,6 +40,7 @@ def main():
36
 
37
  st.session_state["celeb_name"] = st.sidebar.selectbox('Choose a celebrity', options=list(celeb_data.keys()))
38
  model_id=st.sidebar.selectbox("Choose Your Flan-T5 model",options=model_list)
 
39
  st.session_state["QA_model_path"] = f"google/{model_id}" if "flan-t5" in model_id else model_id
40
 
41
  celeb_gender = celeb_data[st.session_state["celeb_name"]]["gender"]
 
14
  st.sidebar.header("CelebChat")
15
  expander = st.sidebar.expander('About the app')
16
  with expander:
17
+ st.markdown("Experience the ultimate celebrity chat demo with this app!")
18
+
19
+ expander = st.sidebar.expander('Disclaimer')
20
+ with expander:
21
+ st.markdown("CelebChat may produce inaccurate information about people, places, or facts.")
22
  if "messages" not in st.session_state:
23
  st.session_state["messages"] = []
24
  if "QA_model_path" not in st.session_state:
 
40
 
41
  st.session_state["celeb_name"] = st.sidebar.selectbox('Choose a celebrity', options=list(celeb_data.keys()))
42
  model_id=st.sidebar.selectbox("Choose Your Flan-T5 model",options=model_list)
43
+
44
  st.session_state["QA_model_path"] = f"google/{model_id}" if "flan-t5" in model_id else model_id
45
 
46
  celeb_gender = celeb_data[st.session_state["celeb_name"]]["gender"]