project_start/conf/dockerfiles/python/scripts/install.py

10 lines
309 B
Python

import nltk
from transformers import MarianMTModel, MarianTokenizer
nltk.download('wordnet')
nltk.download('punkt')
nltk.download('averaged_perceptron_tagger')
model_name = 'Helsinki-NLP/opus-mt-en-ru'
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)