Update app.py
Browse files
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])
|