app.py

import gradio as gr
from transformers import pipeline
from gradio.mix import Parallel

pipe = pipeline("translation", model="t5-base", tokenizer_kwargs={"model_max_length": 1024})

def translate(text, target_language):
    return pipe(text, target_language=target_language)[0]["translation_text"]

def translate_callback(english, language_select):
    target_language = "de" if language_select == "German" else "fr" if language_select == "French" else "lo"
    translated_text = translate(english, target_language)
    return translated_text

def clear_callback(english):
    return ""

def main():
    iface = gr.Interface(
        fn=Parallel(translate_callback),
        inputs=[gr.inputs.Textbox(label="English text")],
        outputs=[gr.outputs.Textbox(label="Translated Text")],
        title="Text Translation",
        description="Translate English text to German, French, or Lao."
    )
    iface.launch()