Fix input line max
Browse files
app.py
CHANGED
@@ -9,7 +9,9 @@ model = T5ForConditionalGeneration.from_pretrained(model_name)
|
|
9 |
def modernisa(inputs):
|
10 |
lines = [line for line in inputs.strip().split('\n') if line.strip()]
|
11 |
text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
|
12 |
-
outputs =
|
|
|
|
|
13 |
return "<br/>".join(output["generated_text"] for output in outputs)
|
14 |
|
15 |
|
@@ -17,12 +19,12 @@ samples = [
|
|
17 |
"""Otra vez, Don Iuan, me dad,
|
18 |
y otras mil vezes los braços.
|
19 |
Otra, y otras mil sean lazos
|
20 |
-
de nuestra antigua amistad.
|
21 |
-
|
22 |
Yo me siento
|
23 |
tan alegre, tan vfano,
|
24 |
-
tan venturoso, tan vano,
|
25 |
-
|
26 |
encareceros jamàs
|
27 |
las venturas que posseo,
|
28 |
porque el pensamiento creo"""
|
@@ -30,6 +32,6 @@ porque el pensamiento creo"""
|
|
30 |
|
31 |
gr.Interface(
|
32 |
fn=modernisa,
|
33 |
-
inputs=gr.inputs.Textbox(lines=
|
34 |
outputs=gr.outputs.HTML(label="Modern Spanish"),
|
35 |
).launch(inline=False)
|
|
|
9 |
def modernisa(inputs):
|
10 |
lines = [line for line in inputs.strip().split('\n') if line.strip()]
|
11 |
text2text_generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
|
12 |
+
outputs = []
|
13 |
+
for line in lines:
|
14 |
+
outputs += text2text_generator([line], max_length=150):
|
15 |
return "<br/>".join(output["generated_text"] for output in outputs)
|
16 |
|
17 |
|
|
|
19 |
"""Otra vez, Don Iuan, me dad,
|
20 |
y otras mil vezes los braços.
|
21 |
Otra, y otras mil sean lazos
|
22 |
+
de nuestra antigua amistad.
|
23 |
+
Como venis?
|
24 |
Yo me siento
|
25 |
tan alegre, tan vfano,
|
26 |
+
tan venturoso, tan vano,
|
27 |
+
que no podrà el pensamiento
|
28 |
encareceros jamàs
|
29 |
las venturas que posseo,
|
30 |
porque el pensamiento creo"""
|
|
|
32 |
|
33 |
gr.Interface(
|
34 |
fn=modernisa,
|
35 |
+
inputs=gr.inputs.Textbox(lines=12, label="Enter Old Spanish", default=random.choice(samples)),
|
36 |
outputs=gr.outputs.HTML(label="Modern Spanish"),
|
37 |
).launch(inline=False)
|