Spaces:

Cricles
/

demo_detoxi

Sleeping

Cricles commited on Apr 9

Commit

2235879

verified ·

1 Parent(s): d02b301

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,8 +35,7 @@ class ModelWrapper(object):
             ModelWrapper.MODELS_DIR + ModelWrapper.MODEL_NAME, torchscript=True
         )
         self.tokenizer = BertTokenizerFast.from_pretrained(
-            str(Path(ModelWrapper.MODELS_DIR + ModelWrapper.TOKENIZER).absolute()),
-            local_files_only=True
         )
         self.id2label: dict[int, str] = {0: "__label__positive", 1: "__label__negative"}

             ModelWrapper.MODELS_DIR + ModelWrapper.MODEL_NAME, torchscript=True
         )
         self.tokenizer = BertTokenizerFast.from_pretrained(
+            "blanchefort/rubert-base-cased-sentiment"
         )
         self.id2label: dict[int, str] = {0: "__label__positive", 1: "__label__negative"}