File size: 826 Bytes
ddc258f 9f52f79 ddc258f 9f52f79 ddc258f 9f52f79 ddc258f 9f52f79 ddc258f 9f52f79 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
from transformers import MarianMTModel, MarianTokenizer
# Load the model and tokenizer
model_name = "Helsinki-NLP/opus-mt-nl-en"
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)
def translate(text):
# Tokenize input text
tokenized_text = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
# Perform translation
translated = model.generate(**tokenized_text)
# Decode translated text
return tokenizer.decode(translated[0], skip_special_tokens=True)
# Define Gradio interface
interface = gr.Interface(fn=translate, inputs="text", outputs="text", title="Dutch to English Translator", description="Translate Dutch text to English using Helsinki-NLP/opus-mt-nl-en.")
# Launch the interface
interface.launch()
|