Spaces:

sebastiansarasti
/

clip_fashion

Running

sebastiansarasti commited on 10 days ago

Commit

a50b87e

•

1 Parent(s): b8c7cfd

solving file path

Files changed (1) hide show

search.py CHANGED Viewed

@@ -28,7 +28,7 @@ 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

 file_path = hf_hub_download(repo_id=model_name, filename=filename)
+clip_model.load_state_dict(torch.load(file_path, map_location=torch.device('cpu')))
 te_final = clip_model.text_encoder
 ie_final = clip_model.image_encoder