from transformers import AutoModelForSeq2SeqLM, AutoTokenizer def create_marian_enfr(): """ Initializes model and tokenizer. """ checkpoint = 'Helsinki-NLP/opus-mt-en-fr' tokenizer = AutoTokenizer.from_pretrained(checkpoint, return_tensors='pt') model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint) return model, tokenizer