Views
No views yet
1from transformers import MarianMTModel, MarianTokenizer
2
3# Load model and tokenizer
4model_name = "playground/opus-mt-de-en"
5tokenizer = MarianTokenizer.from_pretrained(model_name)
6model = MarianMTModel.from_pretrained(model_name)
7
8# Translate German to English
9german_text = "Guten Tag, wie geht es Ihnen?"
10inputs = tokenizer(german_text, return_tensors="pt", padding=True)
11
12# Generate translation
13translated = model.generate(**inputs)
14english_text = tokenizer.decode(translated[0], skip_special_tokens=True)
15
16print(english_text)
17# Output: "Good day, how are you?"1texts = [
2 "Hallo Welt",
3 "Ich liebe maschinelles Lernen",
4 "Das Wetter ist schön heute"
5]
6
7inputs = tokenizer(texts, return_tensors="pt", padding=True)
8translated = model.generate(**inputs)
9
10for i, translation in enumerate(translated):
11 print(f"{texts[i]} -> {tokenizer.decode(translation, skip_special_tokens=True)}")1from transformers import pipeline
2
3translator = pipeline("translation", model="playground/opus-mt-de-en")
4result = translator("Guten Morgen!")
5print(result[0]['translation_text'])