Spaces:
Runtime error
Runtime error
Update recommendation.py
Browse files- 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 |
|