Spaces:
Running
Running
Update model_wrapper/bert_wrapper.py
Browse files
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 |
-
|
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 |
|