Update app.py

app.py

@@ -18,6 +18,8 @@ model = SentenceTransformer('all-mpnet-base-v2')
 flix_ds = load_dataset("nickmuchi/netflix-shows-mpnet-embeddings", use_auth_token=True)
 dataset_embeddings = torch.from_numpy(flix_ds["train"].to_pandas().to_numpy()).to(torch.float)
 
+embed = {'embedding': dataset_embeddings}
+
 #load cross-encoder for reranking
 cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2')
 
@@ -35,7 +37,7 @@ def semantic_search(query,embeddings,top_k):
     '''Encode query and check similarity with embeddings'''
     
     question_embedding = model.encode(query, convert_to_tensor=True).cpu()
-    hits = util.semantic_search(question_embedding, embeddings, top_k=top_k)
+    hits = util.semantic_search(question_embedding, embeddings['embedding'], top_k=top_k)
     hits = hits[0]
     
     ##### Re-Ranking #####
@@ -113,7 +115,7 @@ with demo:
     sem_but = gr.Button('Search')
      
             
-    sem_but.click(semantic_search,inputs=[query,dataset_embeddings,top_k],outputs=[bi_output,cross_output],queue=True)
+    sem_but.click(semantic_search,inputs=[query,embed,top_k],outputs=[bi_output,cross_output],queue=True)
 
     gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=nickmuchi-netflix-shows-semantic-search)")