switched pickle order
Browse files
app.py
CHANGED
@@ -16,8 +16,9 @@ def speech_to_text(speech):
|
|
16 |
|
17 |
bi_encoder = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
|
18 |
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
|
19 |
-
corpus_embeddings=pd.read_pickle("corpus_embeddings_cpu.pkl")
|
20 |
corpus=pd.read_pickle("corpus.pkl")
|
|
|
|
|
21 |
|
22 |
def search(query,top_k=100):
|
23 |
print("Top 3 Answer by the NSE:")
|
|
|
16 |
|
17 |
bi_encoder = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
|
18 |
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
|
|
|
19 |
corpus=pd.read_pickle("corpus.pkl")
|
20 |
+
corpus_embeddings=pd.read_pickle("corpus_embeddings_cpu.pkl")
|
21 |
+
|
22 |
|
23 |
def search(query,top_k=100):
|
24 |
print("Top 3 Answer by the NSE:")
|