lang-uk
/

flair-uk-forward

Text2Text Generation

Model card Files Files and versions Community

Dmitry Chaplinsky commited on Oct 7, 2022

Commit

22e7347

•

1 Parent(s): 062350e

And another one

Files changed (1) hide show

pipeline.py +3 -3

pipeline.py CHANGED Viewed

@@ -6,9 +6,9 @@ class PreTrainedPipeline:
     def __init__(self, path=""):
         from huggingface_hub import hf_hub_download
-        hf_hub_download(repo_id="dchaplinsky/flair-uk-forward", filename="best-lm.pt")
-        self.model = LanguageModel.load_language_model("best-lm.pt")
     def __call__(self, inputs: str) -> List[Dict]:
         """

     def __init__(self, path=""):
         from huggingface_hub import hf_hub_download
+        self.model = LanguageModel.load_language_model(
+            hf_hub_download(repo_id="dchaplinsky/flair-uk-forward", filename="best-lm.pt")
+        )
     def __call__(self, inputs: str) -> List[Dict]:
         """