Logeswaransr commited on
Commit
dc46cad
·
verified ·
1 Parent(s): b512737

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, st.session_state.messages)
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