File size: 1,067 Bytes
68e305f 5c0d878 68e305f 9328f46 68e305f c85bbaa 68e305f 9328f46 c85bbaa 9328f46 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import gradio as gr
from transformers import AutoTokenizer, pipeline, T5ForConditionalGeneration
model_name = "versae/modernisa-pre"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = T5ForConditionalGeneration.from_pretrained(model_name)
def modernisa(inputs):
lines = [line for line in inputs.strip().split('\n') if line.strip()]
text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
outputs = text2text_generator(lines, max_length=150)
return "<br/>".join(output["generated_text"] for output in outputs)
sample = """Otra vez, Don Iuan, me dad,
y otras mil vezes los braços.
Otra, y otras mil sean lazos
de nuestra antigua amistad.
Como venis?
Yo me siento
tan alegre, tan vfano,
tan venturoso, tan vano,
que no podrà el pensamiento
encareceros jamàs
las venturas que posseo,
porque el pensamiento creo"""
gr.Interface(
fn=modernisa,
inputs=gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=sample),
outputs=gr.outputs.HTML(label="Modern Spanish"),
).launch(inline=False) |