|
from transformers import AutoModel,AutoTokenizer,AutoModelForSeq2SeqLM |
|
import gradio as grad |
|
mdl_name = "Helsinki-NLP/opus-mt-en-fr" |
|
mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name) |
|
my_tkn = AutoTokenizer.from_pretrained(mdl_name) |
|
|
|
|
|
|
|
def translate(text): |
|
inputs = my_tkn(text, return_tensors="pt") |
|
trans_output = mdl.generate(**inputs) |
|
response = my_tkn.decode(trans_output[0], skip_special_tokens=True) |
|
|
|
|
|
return response |
|
txt=grad.Textbox(lines=1, label="English", placeholder="English Text here") |
|
out=grad.Textbox(lines=1, label="French") |
|
grad.Interface(translate, inputs=txt, outputs=out).launch() |