versae commited on
Commit
0719bc2
1 Parent(s): e33fddd

Fix input line max

Browse files
Files changed (1) hide show
  1. app.py +8 -6
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 = text2text_generator(lines, max_length=150)
 
 
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
- """Como venis?
22
  Yo me siento
23
  tan alegre, tan vfano,
24
- tan venturoso, tan vano,""",
25
- """que no podrà el pensamiento
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=4, label="Enter Old Spanish", default=random.choice(samples)),
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)