lang-uk
/

fasttext_uk

Feature Extraction

Model card Files Files and versions

dchaplinsky commited on Nov 6, 2022

Commit

eb31495

·

1 Parent(s): 6209f90

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'))
     def __call__(self, inputs: str) -> List[float]:
         """

         """
         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]:
         """