Update recommendation.py

recommendation.py

@@ -3,13 +3,11 @@ from sentence_transformers import SentenceTransformer
 from torchvision import transforms
 from models.encoder import Encoder
 from indexer import Indexer
-import dotenv
 import torch
 import os
 
-dotenv.load_dotenv()
 
-model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+model = SentenceTransformer('intfloat/multilingual-e5-base')
 
 encoder = Encoder()
 encoder.load_state_dict(torch.load('./models/encoder.bin', map_location=torch.device('cpu')))
@@ -25,7 +23,7 @@ image_embedding_index = Indexer(latent_data)
 
 def get_recommendations(image, title):
 #   title = [dataset[product_id]['title']]
-  title_embeds = torch.randn(1,768)#model.encode(title, normalize_embeddings=True)
+  title_embeds = model.encode(title, normalize_embeddings=True)
   image = transforms.ToTensor()(image.convert("L"))
   image_embeds =  encoder(image).detach().numpy()