Initial commit

app.py

@@ -1,13 +1,22 @@
-from transformers import pipeline
-
-model = pipeline("translation", model="vonjack/opus-mt-mul-en-big") 
+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 predict(prompt):
-    completion = model(prompt)[0]["pt"]
-    return completion
+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)
 
-import gradio as gr
+# 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.")
 
-gr.Interface(fn=predict, inputs="text", outputs="text").launch()
+# Launch the interface
+interface.launch()