Skip to content

Commit

Permalink
Model loader uncommented
Browse files Browse the repository at this point in the history
  • Loading branch information
Joffreybvn committed Feb 7, 2021
1 parent 54b8812 commit 89cfa81
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/nlu/classifying/classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 89cfa81

Please sign in to comment.