switching to hf hub

pipeline.py

@@ -1,14 +1,13 @@
-import os.path
 from typing import List, Dict
 from flair.models.language_model import LanguageModel
 
 
 class PreTrainedPipeline:
     def __init__(self, path=""):
-        # IMPLEMENT_THIS
-        # Preload all the elements you are going to need at inference.
-        # For instance your model, processors, tokenizer that might be needed.
-        # This function is only called once, so do all the heavy processing I/O here"""
+        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]: