Yoel125 commited on
Commit
13cabff
·
verified ·
1 Parent(s): 8456386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=10):
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]