Hugging Face's logo

Spaces:
CabraVC
/
holiday_testing
Paused

App Files Community
svystun-taras commited on
Commit
ca57df8
·
1 Parent(s): a61e6e4

fixed a bug

Browse files
Files changed (1) hide show
  1. test_models/create_setfit_model.py +1 -1
test_models/create_setfit_model.py CHANGED
@@ -50,7 +50,7 @@ if __name__ == '__main__' or __name__ == 'create_setfit_model':
50
  load_model(model_head, f'models/linear_head.safetensors')
51
  elif __name__ == 'test_models.create_setfit_model':
52
  model_body = SentenceTransformer('test_models/financial-roberta')
53
- load_model(model_head, f'/test_models/models/linear_head.safetensors')
54
 
55
 
56
  model = SetFitModel(model_body=model_body,
 
50
  load_model(model_head, f'models/linear_head.safetensors')
51
  elif __name__ == 'test_models.create_setfit_model':
52
  model_body = SentenceTransformer('test_models/financial-roberta')
53
+ load_model(model_head, f'/home/user/app/test_models/models/linear_head.safetensors')
54
 
55
 
56
  model = SetFitModel(model_body=model_body,