Update app.py

app.py

@@ -61,12 +61,11 @@ cross_encoder = CrossEncoder(option2, device='cpu')
 passages = []
 
 # load pre-train embeedings files
-embedding_cache_path = 'etsy-embeddings-gpu-total-0530.pkl'
 print("Load pre-computed embeddings from disc")
-with open(embedding_cache_path, "rb") as fIn:
-  cache_data = pickle.load(fIn)
-  passages = cache_data['sentences']
-  corpus_embeddings = cache_data['embeddings']
+embedding_cache_path = 'embeddings.pt'
+corpus_embeddings = torch.load(embedding_cache_path)
+with open('sentences.json', 'r') as file:
+    passages = json.load(file)
 
 from rank_bm25 import BM25Okapi
 from sklearn.feature_extraction import _stop_words