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])
|