HamidBekam commited on
Commit
7eebc27
1 Parent(s): a37eed2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ embedder = SentenceTransformer('sentence-transformers/paraphrase-multilingual-mp
21
 
22
  def get_similar_products(query, products, mean_embeddings_tensor, top_k=10):
23
  query_embedding = embedder.encode(query, convert_to_tensor=True)
24
- cos_scores = util.cos_sim(query_embedding, mean_embeddings_tensor)[0]
25
  top_results = torch.topk(cos_scores, k=top_k)
26
 
27
  similar_products = [(products[idx.item()], score.item()) for score, idx in zip(top_results[0], top_results[1])]
 
21
 
22
  def get_similar_products(query, products, mean_embeddings_tensor, top_k=10):
23
  query_embedding = embedder.encode(query, convert_to_tensor=True)
24
+ cos_scores = util.cos_sim(query_embedding, stored_embeddings)[0]
25
  top_results = torch.topk(cos_scores, k=top_k)
26
 
27
  similar_products = [(products[idx.item()], score.item()) for score, idx in zip(top_results[0], top_results[1])]