Spaces:

GIZ
/

climate_vulnerability_analysis

Sleeping

leavoigt commited on 29 days ago

Commit

8a82291

verified ·

1 Parent(s): 24303fb

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -75,7 +75,7 @@ def load_targetClassifier(config_file:str = None, classifier_name:str = None):
     logging.info("Loading classifier")
     # Loading classifier
-    doc_classifier = SetFitModel.from_pretrained("GIZ/TARGET-VULNERABILITY-multiclass-mpnet)
     return doc_classifier

     logging.info("Loading classifier")
     # Loading classifier
+    doc_classifier = SetFitModel.from_pretrained("GIZ/TARGET-VULNERABILITY-multiclass-mpnet")
     return doc_classifier