Cricles commited on
Commit
f9b0624
·
verified ·
1 Parent(s): bed6434

Update model_wrapper/bert_wrapper.py

Browse files
Files changed (1) hide show
  1. model_wrapper/bert_wrapper.py +1 -1
model_wrapper/bert_wrapper.py CHANGED
@@ -13,7 +13,7 @@ class BertWrapper(object):
13
  BertWrapper.MODELS_DIR + BertWrapper.MODEL_NAME, torchscript=True
14
  )
15
  self.tokenizer = BertTokenizerFast.from_pretrained(
16
- BertWrapper.MODELS_DIR + BertWrapper.TOKENIZER
17
  )
18
  self.id2label: dict[int, str] = {0: "__label__positive", 1: "__label__negative"}
19
 
 
13
  BertWrapper.MODELS_DIR + BertWrapper.MODEL_NAME, torchscript=True
14
  )
15
  self.tokenizer = BertTokenizerFast.from_pretrained(
16
+ "blanchefort/rubert-base-cased-sentiment"
17
  )
18
  self.id2label: dict[int, str] = {0: "__label__positive", 1: "__label__negative"}
19