dchaplinsky commited on
Commit
4ad1834
1 Parent(s): d076f6c

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -8,7 +8,7 @@ class PreTrainedPipeline():
8
  """
9
  Initialize model
10
  """
11
- 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'))
12
 
13
  def __call__(self, inputs: str) -> List[float]:
14
  """
 
8
  """
9
  Initialize model
10
  """
11
+ self.model = fasttext.load_model(os.path.join(path, 'skipgram.uk.300.bin'))
12
 
13
  def __call__(self, inputs: str) -> List[float]:
14
  """