downloading the model from HF

search.py

@@ -1,6 +1,7 @@
 from transformers import ViTModel, AutoModelForMaskedLM, AutoTokenizer, ViTImageProcessor, DistilBertModel
 from pinecone import Pinecone
 import torch
+from huggingface_hub import hf_hub_download
 
 
 pc = Pinecone()
@@ -11,9 +12,6 @@ from io import BytesIO
 import base64
 from PIL import Image
 
-import sys
-
-sys.path.append('../src')
 
 from model import CLIPChemistryModel, TextEncoderHead, ImageEncoderHead
 
@@ -25,7 +23,12 @@ image_encoder = ImageEncoderHead(model=IMAGE_BASE)
 
 clip_model = CLIPChemistryModel(text_encoder=text_encoder, image_encoder=image_encoder)
 
-clip_model.load_state_dict(torch.load('/Users/sebastianalejandrosarastizambonino/Documents/projects/CLIP_Pytorch/src/best_model_fashion.pth', map_location=torch.device('cpu')))
+model_name = "sebastiansarasti/clip_fashion"  
+filename = "best_model.pth"  
+
+file_path = hf_hub_download(repo_id=model_name, filename=filename)
+
+clip_model.load_state_dict(torch.load('best_model_fashion.pth', map_location=torch.device('cpu')))
 
 te_final = clip_model.text_encoder
 ie_final = clip_model.image_encoder