Increase the output tokens

functions.py

@@ -32,7 +32,7 @@ def summarize_text(text: str):
     print(['summarize_text', 'generating'])
     with torch.cuda.amp.autocast():
         output_tokens = model.generate(**batch,
-                                       max_new_tokens=256,
+                                       max_new_tokens=512,
                                        generation_config=generation_config
                                        )
     output = tokenizer.decode(output_tokens[0], skip_special_tokens=True)