Spaces:
Sleeping
Sleeping
Update blueprints/summarize.py
Browse files- 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)
|