Spaces:
Runtime error
Runtime error
lhzstar
commited on
Commit
·
e916883
1
Parent(s):
fb6ade2
new commits
Browse files
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("
|
|
|
|
|
|
|
|
|
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"]
|