Spaces:
Runtime error
Runtime error
Commit
·
5c4c958
1
Parent(s):
35b8bbe
Upload app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ import gradio as gr
|
|
6 |
|
7 |
bi_encoder = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
|
8 |
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
|
9 |
-
corpus_embeddings=pd.read_pickle("
|
10 |
-
corpus=pd.read_pickle("
|
11 |
|
12 |
def search(query,top_k=100):
|
13 |
print("Top 5 Answer by the NSE:")
|
|
|
6 |
|
7 |
bi_encoder = SentenceTransformer("multi-qa-MiniLM-L6-cos-v1")
|
8 |
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
|
9 |
+
corpus_embeddings=pd.read_pickle("corpus_embeddings_cpu.pkl")
|
10 |
+
corpus=pd.read_pickle("corpus.pkl")
|
11 |
|
12 |
def search(query,top_k=100):
|
13 |
print("Top 5 Answer by the NSE:")
|