leavoigt commited on
Commit
ab0f868
·
verified ·
1 Parent(s): 88a3bb2

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +1 -1
utils/target_classifier.py CHANGED
@@ -79,7 +79,7 @@ def load_targetClassifier(config_file:str = None, classifier_name:str = None):
79
  hf_token = os.environ["model_token"]
80
 
81
  # Loading classifier
82
- doc_classifier = SetFitModel.from_pretrained(classifier_name, hf_token)
83
 
84
  return doc_classifier
85
 
 
79
  hf_token = os.environ["model_token"]
80
 
81
  # Loading classifier
82
+ doc_classifier = SetFitModel.from_pretrained(classifier_name, token=hf_token)
83
 
84
  return doc_classifier
85