Another try to deploy

pipeline.py

@@ -1,4 +1,4 @@
-import gensim
+from gensim.models import KeyedVectors
 from typing import List, Dict
 
 
@@ -6,8 +6,9 @@ class PreTrainedPipeline:
     def __init__(self, path=""):
         from huggingface_hub import hf_hub_download
 
-        self.model = gensim.models.Word2Vec.load(
-            hf_hub_download(repo_id="lang-uk/word2vec-uk", filename="ubercorpus.cased.tokenized.300d")
+        self.model = KeyedVectors.load_word2vec_format(
+            hf_hub_download(repo_id="lang-uk/word2vec-uk", filename="ubercorpus.cased.tokenized.300d"),
+            binary=False
         )
 
     def __call__(self, inputs: str) -> List[Dict]:
@@ -19,4 +20,4 @@ class PreTrainedPipeline:
             A :obj:`str`
         """
         inputs = inputs.strip()
-        return [{"generated_text": str(self.model.wv.most_similar(inputs))}]
+        return [{"generated_text": str(self.model.most_similar(inputs, topn=30))}]