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 |
|