# install gradio, transformers and sentencepiece before using this code
from transformers import pipeline
import gradio as gr

model_checkpoint = "rajbhirud/eng-to-fra-model"
translator = pipeline("translation", model=model_checkpoint)

def translation(text):
    return translator(text)[0]['translation_text']

# Create a Gradio interface
iface = gr.Interface(
    fn=translation,
    inputs=gr.inputs.Textbox(label="Input English Text"),
    outputs=gr.outputs.Textbox(label="Translated French Text"),
    title="English to French Translation",
    description="Translate English text to French using a fine-tuned model.",
)

# Launch the interface
iface.launch()