Update blueprints/summarize.py

blueprints/summarize.py

@@ -66,8 +66,14 @@ def run_inference_logic(config):
         }
 
     elif task_type == 'SEQ_2_SEQ_LM':
-        # Summarization specific args
-        out = pipe(text, max_length=512, min_length=30, do_sample=False)
+        out = pipe(
+            text,
+            max_length=512,
+            min_length=30,
+            do_sample=True,
+            temperature=config['temp'],
+            top_k=config['topk']
+        )
         return {"output": out[0]['summary_text']}
 
     else: