Cricles commited on
Commit
e34b55e
·
verified ·
1 Parent(s): 61d66d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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