Update app.py

app.py

@@ -25,8 +25,8 @@ title = "Flan T5 and Vanilla T5"
 description = "Demo that compares [T5-large](https://huggingface.co/t5-large) and [Flan-T5-large](https://huggingface.co/ybelkada/flan-t5-large)"
 
 def inference(text):
-  output_flan = pipe_flan(text)[0]["generated_text"]
-  output_vanilla = pipe_vanilla(text)[0]["generated_text"]
+  output_flan = pipe_flan(text, max_new_tokens=100)[0]["generated_text"]
+  output_vanilla = pipe_vanilla(text, max_new_tokens=100)[0]["generated_text"]
   return [output_flan, output_vanilla]
 
 io = gr.Interface(