broadfield-dev commited on
Commit
2cd2e26
·
verified ·
1 Parent(s): 952b0ec

Update blueprints/summarize.py

Browse files
Files changed (1) hide show
  1. blueprints/summarize.py +1 -1
blueprints/summarize.py CHANGED
@@ -92,7 +92,7 @@ def api_summarize():
92
  config = {
93
  "text": data['text'],
94
  "model_name": data.get('model_name', "facebook/bart-large-cnn"),
95
- "hf_token": data.get('hf_token', hf_token)
96
  # We force this for the specific summarization UI,
97
  # but the backend logic supports others.
98
  "task_type": "SEQ_2_SEQ_LM"
 
92
  config = {
93
  "text": data['text'],
94
  "model_name": data.get('model_name', "facebook/bart-large-cnn"),
95
+ "hf_token": data.get('hf_token', hf_token),
96
  # We force this for the specific summarization UI,
97
  # but the backend logic supports others.
98
  "task_type": "SEQ_2_SEQ_LM"