Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,8 @@ class ModelWrapper(object):
|
|
34 |
ModelWrapper.MODELS_DIR + ModelWrapper.MODEL_NAME, torchscript=True
|
35 |
)
|
36 |
self.tokenizer = BertTokenizerFast.from_pretrained(
|
37 |
-
ModelWrapper.MODELS_DIR + ModelWrapper.TOKENIZER
|
|
|
38 |
)
|
39 |
self.id2label: dict[int, str] = {0: "__label__positive", 1: "__label__negative"}
|
40 |
|
|
|
34 |
ModelWrapper.MODELS_DIR + ModelWrapper.MODEL_NAME, torchscript=True
|
35 |
)
|
36 |
self.tokenizer = BertTokenizerFast.from_pretrained(
|
37 |
+
ModelWrapper.MODELS_DIR + ModelWrapper.TOKENIZER,
|
38 |
+
local_files_only=True
|
39 |
)
|
40 |
self.id2label: dict[int, str] = {0: "__label__positive", 1: "__label__negative"}
|
41 |
|