bhulston commited on
Commit
50b1dd2
·
1 Parent(s): 845068b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -97,13 +97,12 @@ def get_rag_results(prompt):
97
  )
98
 
99
  response, additional_metadata = clean_pinecone(response)
100
- print("Here is the pinecone response:", response)
101
  if len(response) < 1:
102
  response = "No classes were found that matched your criteria"
103
  additional_metadata = "None"
104
  else:
105
  response = reranker(query, response) # BERT cross encoder for ranking
106
- print("##Here is the response from the reranker", response)
107
 
108
 
109
  return response, additional_metadata
@@ -136,6 +135,7 @@ if prompt := st.chat_input("What kind of class are you looking for?"):
136
  rag_response, additional_metadata = get_rag_results(prompt)
137
  rag_response = " ".join([message for message in rag_response])
138
  st.session_state.rag_responses.append(rag_response)
 
139
  all_rag_responses = " ".join([response for response in st.session_state.rag_responses])
140
  result_query = 'Original Query:' + prompt
141
  # '\n Additional Class Times:' + str(additional_metadata)
 
97
  )
98
 
99
  response, additional_metadata = clean_pinecone(response)
 
100
  if len(response) < 1:
101
  response = "No classes were found that matched your criteria"
102
  additional_metadata = "None"
103
  else:
104
  response = reranker(query, response) # BERT cross encoder for ranking
105
+ print("## Here is the response from the reranker", response)
106
 
107
 
108
  return response, additional_metadata
 
135
  rag_response, additional_metadata = get_rag_results(prompt)
136
  rag_response = " ".join([message for message in rag_response])
137
  st.session_state.rag_responses.append(rag_response)
138
+ print("Here is the session state responses", st.session_state.rag_responses)
139
  all_rag_responses = " ".join([response for response in st.session_state.rag_responses])
140
  result_query = 'Original Query:' + prompt
141
  # '\n Additional Class Times:' + str(additional_metadata)