diff --git a/src/nlu/classifying/classifier.py b/src/nlu/classifying/classifier.py index ab63a37..26c0278 100644 --- a/src/nlu/classifying/classifier.py +++ b/src/nlu/classifying/classifier.py @@ -51,7 +51,7 @@ def _load_model(self) -> BertForSequenceClassification: """ # Download and save the weights locally - # self.__load_remote_file(config.MODEL_WEIGHT_URL, config.MODEL_WEIGHT_LOCAL_COPY) + self.__load_remote_file(config.MODEL_WEIGHT_URL, config.MODEL_WEIGHT_LOCAL_COPY) # Instantiate the model model = BertForSequenceClassification.from_pretrained(