Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,14 +5,14 @@ Created on Sun Oct 01 10:49:43 2023
|
|
5 |
|
6 |
import streamlit as st
|
7 |
import sentencepiece
|
8 |
-
from interaction import generate_response, audio_response
|
9 |
|
10 |
st.set_page_config(page_title='AI Chaperone - RAG', layout='wide')
|
11 |
|
12 |
if 'messages' not in st.session_state:
|
13 |
st.session_state.messages=[]
|
14 |
|
15 |
-
|
16 |
st.subheader("AI Chaperone")
|
17 |
|
18 |
for message in st.session_state.messages:
|
@@ -27,7 +27,7 @@ if prompt:=st.chat_input("Enter your query"):
|
|
27 |
'role':'user',
|
28 |
'content': prompt})
|
29 |
|
30 |
-
response=generate_response(prompt,
|
31 |
with st.chat_message("assistant"):
|
32 |
st.markdown(response)
|
33 |
|
|
|
5 |
|
6 |
import streamlit as st
|
7 |
import sentencepiece
|
8 |
+
from interaction import generate_response, audio_response, init_pipeline
|
9 |
|
10 |
st.set_page_config(page_title='AI Chaperone - RAG', layout='wide')
|
11 |
|
12 |
if 'messages' not in st.session_state:
|
13 |
st.session_state.messages=[]
|
14 |
|
15 |
+
pipeline = init_pipeline()
|
16 |
st.subheader("AI Chaperone")
|
17 |
|
18 |
for message in st.session_state.messages:
|
|
|
27 |
'role':'user',
|
28 |
'content': prompt})
|
29 |
|
30 |
+
response=generate_response(prompt, pipeline)
|
31 |
with st.chat_message("assistant"):
|
32 |
st.markdown(response)
|
33 |
|