Update app.py

app.py

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