Ransaka commited on
Commit
b03a557
1 Parent(s): 20ab731

Update recommendation.py

Browse files
Files changed (1) hide show
  1. recommendation.py +1 -1
recommendation.py CHANGED
@@ -36,7 +36,7 @@ def get_recommendations(image, title, k):
36
  title_embeds = model.encode([title], normalize_embeddings=True)
37
  image = transforms.ToTensor()(image.convert("L"))
38
  image_embeds = encoder(image).detach().numpy()
39
- image_embeds = image_embeds / np.linalg.norm(image_embeds, axis=1, keepdims=True)
40
  final_embeds = np.concatenate((image_embeds,title_embeds), axis=1)
41
 
42
  # candidates = index.topk(final_embeds,k=k)
 
36
  title_embeds = model.encode([title], normalize_embeddings=True)
37
  image = transforms.ToTensor()(image.convert("L"))
38
  image_embeds = encoder(image).detach().numpy()
39
+ # image_embeds = image_embeds / np.linalg.norm(image_embeds, axis=1, keepdims=True)
40
  final_embeds = np.concatenate((image_embeds,title_embeds), axis=1)
41
 
42
  # candidates = index.topk(final_embeds,k=k)