Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ dataset_embeddings = torch.tensor(embeddings, device='cpu')
|
|
| 24 |
model = SentenceTransformer('sentence-transformers/clip-ViT-B-32', device='cpu')
|
| 25 |
translator = GoogleTranslator(source='auto', target='en')
|
| 26 |
|
| 27 |
-
def get_recommendations(text_input, top_k=
|
| 28 |
query_embedding = model.encode(text_input, convert_to_tensor=True, device='cpu')
|
| 29 |
similarities = util.cos_sim(query_embedding, dataset_embeddings)[0]
|
| 30 |
top_indices = similarities.argsort(descending=True)[:top_k]
|
|
|
|
| 24 |
model = SentenceTransformer('sentence-transformers/clip-ViT-B-32', device='cpu')
|
| 25 |
translator = GoogleTranslator(source='auto', target='en')
|
| 26 |
|
| 27 |
+
def get_recommendations(text_input, top_k=3):
|
| 28 |
query_embedding = model.encode(text_input, convert_to_tensor=True, device='cpu')
|
| 29 |
similarities = util.cos_sim(query_embedding, dataset_embeddings)[0]
|
| 30 |
top_indices = similarities.argsort(descending=True)[:top_k]
|