Update blueprints/summarize.py

blueprints/summarize.py

@@ -71,8 +71,8 @@ def run_inference_logic(config):
             max_length=512,
             min_length=30,
             do_sample=True,
-            temperature=config['temp'],
-            top_k=config['topk']
+            temperature=float(config['temp']),
+            top_k=int(config['topk'])
         )
         return {"output": out[0]['summary_text']}
 
@@ -99,6 +99,8 @@ def api_summarize():
         "text": data['text'],
         "model_name": data.get('model_name', "facebook/bart-large-cnn"),
         "hf_token": data.get('hf_token', hf_token),
+        "temp": data.get('temp','0.7'),
+        "topk": data.get('topk','50'),
         # We force this for the specific summarization UI, 
         # but the backend logic supports others.
         "task_type": "SEQ_2_SEQ_LM"