bhulston commited on
Commit
c1e65f1
1 Parent(s): 9faf703

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -101,9 +101,7 @@ def get_rag_results(prompt):
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
109
 
@@ -131,6 +129,7 @@ if prompt := st.chat_input("What kind of class are you looking for?"):
131
  messages = [{"role": m["role"], "content": m["content"]}
132
  for m in st.session_state.messages[-6:]]
133
  message_history = " ".join([message["content"] for message in messages])
 
134
 
135
  rag_response, additional_metadata = get_rag_results(prompt)
136
  rag_response = " ".join([message for message in rag_response])
 
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
 
106
  return response, additional_metadata
107
 
 
129
  messages = [{"role": m["role"], "content": m["content"]}
130
  for m in st.session_state.messages[-6:]]
131
  message_history = " ".join([message["content"] for message in messages])
132
+ print("Prompt is", prompt)
133
 
134
  rag_response, additional_metadata = get_rag_results(prompt)
135
  rag_response = " ".join([message for message in rag_response])