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
|