File size: 665 Bytes
f2e64c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# 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() |