Commit
·
e2742b6
1
Parent(s):
6072be8
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -23,7 +23,7 @@ class PreTrainedPipeline():
|
|
| 23 |
- "score": A score between 0 and 1 describing how confident the model is for this label/class.
|
| 24 |
"""
|
| 25 |
# pop inputs for pipeline
|
| 26 |
-
inputs =
|
| 27 |
parameters = data.pop("parameters", None)
|
| 28 |
|
| 29 |
# pass inputs with all kwargs in data
|
|
|
|
| 23 |
- "score": A score between 0 and 1 describing how confident the model is for this label/class.
|
| 24 |
"""
|
| 25 |
# pop inputs for pipeline
|
| 26 |
+
inputs = data.pop("inputs", data)
|
| 27 |
parameters = data.pop("parameters", None)
|
| 28 |
|
| 29 |
# pass inputs with all kwargs in data
|