binqiangliu commited on
Commit
042e63c
·
1 Parent(s): 6831b20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -175,6 +175,7 @@ with st.sidebar:
175
  st.session_state.texts = st.session_state.temp_texts
176
  st.session_state.initial_embeddings=get_embeddings(st.session_state.texts)
177
  st.session_state.db_embeddings = torch.FloatTensor(st.session_state.initial_embeddings)
 
178
  except Exception as e:
179
  # st.write("Unknow error.")
180
  # print("Please enter a valide URL.")
@@ -186,6 +187,7 @@ if st.button('Get AI Response'):
186
  with st.spinner("AI Thinking...Please wait a while to Cheers!"):
187
  q_embedding=get_embeddings(user_question)
188
  final_q_embedding = torch.FloatTensor(q_embedding)
 
189
  hits = semantic_search(final_q_embedding, st.session_state.db_embeddings, top_k=5)
190
  page_contents = []
191
  for i in range(len(hits[0])):
 
175
  st.session_state.texts = st.session_state.temp_texts
176
  st.session_state.initial_embeddings=get_embeddings(st.session_state.texts)
177
  st.session_state.db_embeddings = torch.FloatTensor(st.session_state.initial_embeddings)
178
+ print("DB Embeddings Ready.")
179
  except Exception as e:
180
  # st.write("Unknow error.")
181
  # print("Please enter a valide URL.")
 
187
  with st.spinner("AI Thinking...Please wait a while to Cheers!"):
188
  q_embedding=get_embeddings(user_question)
189
  final_q_embedding = torch.FloatTensor(q_embedding)
190
+ print("Question Embeddings Ready.")
191
  hits = semantic_search(final_q_embedding, st.session_state.db_embeddings, top_k=5)
192
  page_contents = []
193
  for i in range(len(hits[0])):