Update pipeline.py
Browse files- pipeline.py +2 -0
pipeline.py
CHANGED
|
@@ -53,4 +53,6 @@ class PreTrainedPipeline():
|
|
| 53 |
preds = self.tokenizer.batch_decode(output_ids, skip_special_tokens=True)
|
| 54 |
preds = [pred.strip() for pred in preds]
|
| 55 |
|
|
|
|
|
|
|
| 56 |
return preds
|
|
|
|
| 53 |
preds = self.tokenizer.batch_decode(output_ids, skip_special_tokens=True)
|
| 54 |
preds = [pred.strip() for pred in preds]
|
| 55 |
|
| 56 |
+
preds = [{"label": preds[0], "score": 1.0}]
|
| 57 |
+
|
| 58 |
return preds
|