cached_file <- hf_bucket_url

README.md

@@ -50,11 +50,11 @@ class TransformersUD(object):
     if os.path.isdir(bert):
       d,t=x(os.path.join(bert,"deprel")),x(os.path.join(bert,"tagger"))
     else:
-      from transformers.file_utils import hf_bucket_url
-      c=AutoConfig.from_pretrained(hf_bucket_url(bert,"deprel/config.json"))
-      d=x(hf_bucket_url(bert,"deprel/pytorch_model.bin"),config=c)
-      s=AutoConfig.from_pretrained(hf_bucket_url(bert,"tagger/config.json"))
-      t=x(hf_bucket_url(bert,"tagger/pytorch_model.bin"),config=s)
+      from transformers.utils import cached_file
+      c=AutoConfig.from_pretrained(cached_file(bert,"deprel/config.json"))
+      d=x(cached_file(bert,"deprel/pytorch_model.bin"),config=c)
+      s=AutoConfig.from_pretrained(cached_file(bert,"tagger/config.json"))
+      t=x(cached_file(bert,"tagger/pytorch_model.bin"),config=s)
     self.deprel=TokenClassificationPipeline(model=d,tokenizer=self.tokenizer,
       aggregation_strategy="simple")
     self.tagger=TokenClassificationPipeline(model=t,tokenizer=self.tokenizer)