Ah fml

pipeline.py

@@ -7,8 +7,7 @@ class PreTrainedPipeline:
         from huggingface_hub import hf_hub_download
 
         self.model = KeyedVectors.load_word2vec_format(
-            hf_hub_download(repo_id="lang-uk/word2vec-uk", filename="ubercorpus.cased.tokenized.300d"),
-            binary=False
+            hf_hub_download(repo_id="lang-uk/word2vec-uk", filename="ubercorpus.cased.tokenized.300d"), binary=False
         )
 
     def __call__(self, inputs: str) -> List[Dict]:
@@ -20,4 +19,4 @@ class PreTrainedPipeline:
             A :obj:`str`
         """
         inputs = inputs.strip()
-        return [{"generated_text": "\n".join(f"{k}\t{v}" for k, v in self.model.most_similar(inputs, topn=30)))}]
+        return [{"generated_text": "\n".join(f"{k}\t{v}" for k, v in self.model.most_similar(inputs, topn=30))}]