Sabareeshr commited on
Commit
fb9908d
·
1 Parent(s): f598ca0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("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:")
 
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("https://huggingface.co/Sabareeshr/neural-search-engine/resolve/main/corpus_embeddings_cpu.pkl")
10
+ corpus=pd.read_pickle("https://huggingface.co/Sabareeshr/neural-search-engine/resolve/main/corpus.pkl")
11
 
12
  def search(query,top_k=100):
13
  print("Top 5 Answer by the NSE:")