Spaces:
Runtime error
Runtime error
from transformers import T5ForConditionalGeneration, T5Tokenizer | |
import gradio as grad | |
text2text_tkn= T5Tokenizer.from_pretrained("t5-small") | |
mdl = T5ForConditionalGeneration.from_pretrained("t5-small") | |
def text2text_translation(text): | |
inp = "translate English to German:: "+text | |
enc = text2text_tkn(inp, return_tensors="pt") | |
tokens = mdl.generate(**enc) | |
response=text2text_tkn.batch_decode(tokens) | |
return response | |
para=grad.Textbox(lines=1, label="English Text", placeholder="Text in English") | |
out=grad.Textbox(lines=1, label="German Translation") | |
grad.Interface(text2text_translation, inputs=para, | |
outputs=out).launch() |