Spaces:

GIZ
/

climate_vulnerability_analysis

Sleeping

leavoigt commited on 10 days ago

Commit

fd41e66

verified ·

1 Parent(s): 83f4ec2

Update utils/vulnerability_classifier.py

Files changed (1) hide show

utils/vulnerability_classifier.py CHANGED Viewed

@@ -83,7 +83,7 @@ def load_vulnerabilityClassifier(config_file:str = None, classifier_name:str = N
     hf_token = os.environ["model_token"]
     # Download model from HF Hub
-    doc_classifier = SetFitModel.from_pretrained(classifier_name, model_token=hf_token)
     return doc_classifier

     hf_token = os.environ["model_token"]
     # Download model from HF Hub
+    doc_classifier = SetFitModel.from_pretrained(classifier_name, token=hf_token)
     return doc_classifier