Update app.py
Browse files
app.py
CHANGED
|
@@ -157,7 +157,6 @@ def search_func(query):
|
|
| 157 |
|
| 158 |
|
| 159 |
# Encode the query using the bi-encoder and find relevant answers
|
| 160 |
-
corpus_embeddings = pd.read_pickle("corpus_embeddings_cpu.pkl")
|
| 161 |
question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
|
| 162 |
question_embedding = question_embedding.cpu()
|
| 163 |
hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=2, score_function=util.dot_score)
|
|
|
|
| 157 |
|
| 158 |
|
| 159 |
# Encode the query using the bi-encoder and find relevant answers
|
|
|
|
| 160 |
question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
|
| 161 |
question_embedding = question_embedding.cpu()
|
| 162 |
hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=2, score_function=util.dot_score)
|