Spaces:
Running
Running
HamidBekam
commited on
Commit
•
7eebc27
1
Parent(s):
a37eed2
Update app.py
Browse files
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,
|
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])]
|