bhulston commited on
Commit
5490950
1 Parent(s): 26ed99e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -97,8 +97,13 @@ def get_rag_results(prompt):
97
 
98
  response, additional_metadata = clean_pinecone(response)
99
  print("Here is the pinecone response:", response)
100
- response = reranker(query, response) # BERT cross encoder for ranking
101
- print("Here is the response from the reranker", response)
 
 
 
 
 
102
 
103
  return response, additional_metadata
104
 
 
97
 
98
  response, additional_metadata = clean_pinecone(response)
99
  print("Here is the pinecone response:", 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
109