import gradio as gr
from transformers import MarianMTModel, MarianTokenizer

# Load the tokenizer and model from Hugging Face's Transformers library
model_name = 'Helsinki-NLP/opus-mt-en-fr'
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

# Function to translate text from English to French
def translate(text):
    # Tokenize the text using the __call__ method
    model_inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
    
    # Perform the translation
    gen = model.generate(**model_inputs)
    
    # Decode the generated tokens to string
    translation = tokenizer.batch_decode(gen, skip_special_tokens=True)
    return translation[0]

# Create a Gradio interface
iface = gr.Interface(
    fn=translate, 
    inputs=gr.Textbox(lines=2, placeholder="Enter Text in English..."), 
    outputs=gr.Textbox()
)


iface.launch(debug=True,inline=False)