Update accessibility_classifier/classifier/model.py

accessibility_classifier/classifier/model.py

@@ -59,8 +59,7 @@ class Model:
             "cuda:0" if torch.cuda.is_available() else "cpu")
 
         self.labels = ['Accessibility', 'Non-accessibility']
-        self.tokenizer = AlbertTokenizer.from_pretrained(
-            'albert-base-v2', do_lower_case=True)
+        self.tokenizer = BertTokenizer.from_pretrained('bert-base-uncased', do_lower_case=True)
         classifier = AlbertForMultilabelSequenceClassification.from_pretrained(
             'albert-base-v2',
             output_attentions=False,