shrink

ud.py

@@ -1,5 +1,4 @@
 from transformers import TokenClassificationPipeline
-from transformers.pipelines import PIPELINE_REGISTRY
 
 class UniversalDependenciesPipeline(TokenClassificationPipeline):
   def _forward(self,model_input):
@@ -31,5 +30,3 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
       q=self.model.config.id2label[p[i,h[i]]].split("|")
       u+="\t".join([str(i),t[s:e],"_",q[0],"_","|".join(q[1:-1]),str(h[i]),q[-1],"_","_" if i<len(v) and e<v[i][0] else "SpaceAfter=No"])+"\n"
     return u+"\n"
-
-PIPELINE_REGISTRY.register_pipeline("universal-dependencies",pipeline_class=UniversalDependenciesPipeline)