Spaces:

jeevavijay10
/

nlp-goemotions-senti-pred

Runtime error

cspocketindia commited on May 14, 2023

Commit

274f98b

•

1 Parent(s): eb76698

revert tokenizer

Files changed (1) hide show

src/models/bert.py CHANGED Viewed

@@ -20,8 +20,8 @@ class BERTClassifier():
         self.model_name = model_name
-        self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
-        # self.tokenizer = BertTokenizer.from_pretrained(tokenizer_name, do_lower_case=True)
         if model_name.startswith('jeevavijay10'):
             # self.model = torch.load(model_name)

         self.model_name = model_name
+        # self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
+        self.tokenizer = BertTokenizer.from_pretrained(tokenizer_name, do_lower_case=True)
         if model_name.startswith('jeevavijay10'):
             # self.model = torch.load(model_name)