Update app.py

app.py

@@ -17,7 +17,7 @@ model.to(device)
 
 def generate_text(prompt):
     input_ids = tokenizer(prompt, return_tensors="pt").input_ids
-    tokens = model.generate(input_ids, max_length=20)
+    tokens = model.generate(input_ids, max_length=22)
     return tokenizer.decode(tokens[0].tolist(), skip_special_tokens=True)
 
 iface = gr.Interface(