dchaplinsky
commited on
Commit
•
eb31495
1
Parent(s):
6209f90
Update pipeline.py
Browse files- 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'))
|
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, '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 |
"""
|