commit

app.py

@@ -551,14 +551,12 @@ def upscaler(
     return enhanced_image
 
 def summarize_text(
-    text, max_length=10
+    text, max_length=16
 ):
     log(f'CALL summarize_text')
     summary = pegasus_tokenizer.decode( pegasus_model.generate(
         pegasus_tokenizer(text,return_tensors="pt").input_ids,
-        max_length=max_length,
-        num_beams = 4,
-        early_stopping=True
+        max_length=max_length
     )[0], skip_special_tokens=True)
     log(f'RET summarize_text with summary as {summary}')
     return summary