diff options
| author | ZJaume <jzaragoza@prompsit.com> | 2022-07-20 16:35:16 +0300 |
|---|---|---|
| committer | Jaume Zaragoza <ZJaume@users.noreply.github.com> | 2022-07-27 15:20:55 +0300 |
| commit | 516da3fdcb5ff85e9dc16b96b9498dca21a68398 (patch) | |
| tree | 862bb5e5451c4e3149f781399bd57430b74ba342 | |
| parent | 7d241321c3a7f40631694329f30a11017ea15457 (diff) | |
Remove vocab and model file config values, use only dir
| -rw-r--r-- | bicleaner_ai/models.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/bicleaner_ai/models.py b/bicleaner_ai/models.py index 75dd80a..6c3db5d 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -486,8 +486,6 @@ class BCXLMRoberta(BaseModel): self.tokenizer = None self.settings = { - "model_file": "model.tf", - "vocab_file": "vocab", "model": 'jplu/tf-xlm-roberta-base', "batch_size": 16, "maxlen": 150, @@ -537,9 +535,8 @@ class BCXLMRoberta(BaseModel): def load(self): ''' Load fine-tuned model ''' - vocab_file = self.dir + '/' + self.settings["vocab_file"] - self.tokenizer = XLMRobertaTokenizerFast.from_pretrained(vocab_file) - self.model = self.load_model(self.dir+'/'+self.settings["model_file"]) + self.tokenizer = XLMRobertaTokenizerFast.from_pretrained(self.dir) + self.model = self.load_model(self.dir) def softmax_pos_prob(self, x): # Compute softmax probability of the second (positive) class |