broadfield-dev commited on
Commit
fe43ce2
·
verified ·
1 Parent(s): d912461

Update blueprints/summarize.py

Browse files
Files changed (1) hide show
  1. blueprints/summarize.py +1 -1
blueprints/summarize.py CHANGED
@@ -52,7 +52,7 @@ def run_inference_logic(config):
52
  text = config['text']
53
  task_type = config['task_type']
54
  hf_token = config['hf_token']
55
- pipe = get_pipeline(model_id, task_type)
56
 
57
  if task_type == 'TOKEN_CLS':
58
  results = pipe(text)
 
52
  text = config['text']
53
  task_type = config['task_type']
54
  hf_token = config['hf_token']
55
+ pipe = get_pipeline(model_id, task_type, hf_token)
56
 
57
  if task_type == 'TOKEN_CLS':
58
  results = pipe(text)