osanseviero commited on
Commit
911bd1c
1 Parent(s): 263d9d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ title = "Flan T5 and Vanilla T5"
25
  description = "Demo that compares [T5-large](https://huggingface.co/t5-large) and [Flan-T5-large](https://huggingface.co/ybelkada/flan-t5-large)"
26
 
27
  def inference(text):
28
- output_flan = pipe_flan(text, max_new_tokens=100)[0]["generated_text"]
29
- output_vanilla = pipe_vanilla(text, max_new_tokens=100)[0]["generated_text"]
30
  return [output_flan, output_vanilla]
31
 
32
  io = gr.Interface(
 
25
  description = "Demo that compares [T5-large](https://huggingface.co/t5-large) and [Flan-T5-large](https://huggingface.co/ybelkada/flan-t5-large)"
26
 
27
  def inference(text):
28
+ output_flan = pipe_flan(text, max_length=100)[0]["generated_text"]
29
+ output_vanilla = pipe_vanilla(text, max_length=100)[0]["generated_text"]
30
  return [output_flan, output_vanilla]
31
 
32
  io = gr.Interface(