lang-uk
/

fasttext_uk

Feature Extraction

Model card Files Files and versions Community

dchaplinsky commited on Nov 6, 2022

Commit

4ad1834

·

1 Parent(s): d076f6c

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -8,7 +8,7 @@ class PreTrainedPipeline():
         """
         Initialize model
         """
-        self.model = fasttext.load_model(os.path.join(path, 'ubertext.fiction_news_wikipedia.filter_rus+short.tokens.txt.algo-skipgram.epochs-15.subwords-2..5.wordngram-3.neg_sampling-15.bin'))
     def __call__(self, inputs: str) -> List[float]:
         """

         """
         Initialize model
         """
+        self.model = fasttext.load_model(os.path.join(path, 'skipgram.uk.300.bin'))
     def __call__(self, inputs: str) -> List[float]:
         """