TransformersSlowUD

README.md

@@ -59,7 +59,7 @@ class TaggerPipeline(TokenClassificationPipeline):
           e=([x for x in a2b[i+1:] if x>[]]+[[len(text)]])[0][0]
         d[i]["start"],d[i]["end"]=s,e
     return d
-class TransformersUD(object):
+class TransformersSlowUD(object):
   def __init__(self,bert):
     import os
     self.tokenizer=AutoTokenizer.from_pretrained(bert)
@@ -105,7 +105,7 @@ class TransformersUD(object):
         str(h[i]),p,"_","_" if i<n and w[i][0]<e else "SpaceAfter=No"])+"\n"
     return u+"\n"
 
-nlp=TransformersUD("KoichiYasuoka/deberta-large-japanese-unidic-ud-head")
+nlp=TransformersSlowUD("KoichiYasuoka/deberta-large-japanese-unidic-ud-head")
 print(nlp("全学年にわたって小学校の国語の教科書に挿し絵が用いられている"))
 ```