Update app.py

app.py

@@ -9,7 +9,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
 # define a predict function to generate text
 def generate_text(prompt):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
-    generated_text = model.generate(input_ids, max_length=999999, top_p=0.9, top_k=40)
+    generated_text = model.generate(input_ids, max_length=100, top_p=0.9, top_k=40)
     text = tokenizer.decode(generated_text[0], skip_special_tokens=True)
     return text