Spaces:

malmukhtar
/

Accessibility-labeler

Sleeping

malmukhtar commited on Aug 1, 2023

Commit

53ed89d

1 Parent(s): 65d7dba

Update accessibility_classifier/classifier/model.py

Files changed (1) hide show

accessibility_classifier/classifier/model.py CHANGED Viewed

@@ -1,5 +1,5 @@
 from transformers.modeling_outputs import SequenceClassifierOutput
-from transformers import AlbertForSequenceClassification, AlbertTokenizer
 import torch
 import torch.nn.functional as F
 import numpy as np

 from transformers.modeling_outputs import SequenceClassifierOutput
+from transformers import AlbertForSequenceClassification, BertTokenizer
 import torch
 import torch.nn.functional as F
 import numpy as np