# prompt: write a streamlit app that converts english text into french text when translate button is pressed. The title of page should be "Translate to French"."
from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
import streamlit as st 

st.title("Translate to French")

input_text = st.text_input("Enter text to translate")

if st.button("Translate"):
    # Load the pre-trained model and tokenizer
    model_name = "facebook/m2m100_418M"
    model = M2M100ForConditionalGeneration.from_pretrained(model_name)
    tokenizer = M2M100Tokenizer.from_pretrained(model_name)

    # Tokenize input text
    input_ids = tokenizer.encode(input_text, return_tensors="pt")

    # Generate translation
    translated_ids = model.generate(input_ids, target_language="fr")

    # Decode and display the translated text
    translation = tokenizer.decode(translated_ids[0], skip_special_tokens=True)
    st.write(translation)