Spaces:

sebastiansarasti
/

clip_fashion

Runtime error

sebastiansarasti commited on Dec 17, 2024

Commit

a50b87e

verified ·

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