Ransaka commited on
Commit
f05764b
·
verified ·
1 Parent(s): 51fb370

Update recommendation.py

Browse files
Files changed (1) hide show
  1. recommendation.py +1 -1
recommendation.py CHANGED
@@ -23,7 +23,7 @@ image_embedding_index = Indexer(latent_data)
23
 
24
  def get_recommendations(image, title):
25
  # title = [dataset[product_id]['title']]
26
- title_embeds = model.encode(title, normalize_embeddings=True)
27
  image = transforms.ToTensor()(image.convert("L"))
28
  image_embeds = encoder(image).detach().numpy()
29
 
 
23
 
24
  def get_recommendations(image, title):
25
  # title = [dataset[product_id]['title']]
26
+ title_embeds = model.encode([title], normalize_embeddings=True)
27
  image = transforms.ToTensor()(image.convert("L"))
28
  image_embeds = encoder(image).detach().numpy()
29