Update app.py

app.py

@@ -14,7 +14,7 @@ def predict(name, description):
     text = 'Description: '+ description + '; Name: ' + name 
     query_embedding = model.encode(text, convert_to_tensor=True)
     
-    corpus_embeddings = torch.Tensor(corpus.iloc[:,5])
+    corpus_embeddings = torch.Tensor(corpus["embeddings"])
 
     output = sentence_transformers.util.semantic_search(query_embedding, corpus_embeddings, top_k = 5)