TRying to deploy

README.md

@@ -1,3 +1,14 @@
 ---
+language:
+  - uk
+tags:
+  - text2text-generation
+  - gensim
+library_name: generic
 license: mit
+datasets:
+- ubertext1.0
+widget:
+- text: "Подорож"
+- text: "Голова"
 ---

pipeline.py

@@ -0,0 +1,22 @@
+import gensim
+from typing import List, Dict
+
+
+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")
+        )
+
+    def __call__(self, inputs: str) -> List[Dict]:
+        """
+        Args:
+            inputs (:obj:`str`):
+                a string containing some text
+        Return:
+            A :obj:`str`
+        """
+        inputs = inputs.strip()
+        return [{"generated_text": str(self.model.wv.most_similar(inputs))}]

requirements.txt

@@ -0,0 +1 @@
+gensim