app.py

import gradio as gr
from transformers import pipeline

# Define the translation pipeline
translator = pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt")

# Define the Gradio interface
def translate_text(text, target_language):
    translated_text = translator(text, target_language)[0]['translation_text']
    return translated_text

iface = gr.Interface(
    fn=translate_text,
    inputs=[
        gr.inputs.Textbox(label="Input Text"),
        gr.inputs.Dropdown(
            label="Target Language",
            choices=["es", "fr", "de", "it"],  # Add more languages as needed
        ),
    ],
    outputs=gr.outputs.Textbox(label="Translated Text"),
    live=True,
    title="Multilingual Translation Tool",
    description="Translate text into multiple languages using facebook/mbart-large-50-many-to-many-mmt model.",
)

# Start the Gradio interface
iface.launch()