Update app.py

app.py

@@ -10,7 +10,7 @@ def generate_text(prompt):
  input_ids = tokenizer.encode(prompt, return_tensors="pt")
  attention_mask = torch.ones(input_ids.shape, dtype=torch.long)
 
- max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 100
+ max_length = model.config.n_positions if len(input_ids[0]) > model.config.n_positions else len(input_ids[0]) + 50
  beam_output = model.generate(
  input_ids,
  attention_mask=attention_mask,