ydshieh
/
vit-gpt2-coco-en
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update pipeline.py

This commit is contained in:
Yih-Dar SHIEH 2021-10-25 08:19:08 +00:00 committed by huggingface-web
parent 117d46a115
commit 45bde45ae6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pipeline.py
View File

@ -53,4 +53,6 @@ class PreTrainedPipeline():
preds = self.tokenizer.batch_decode(output_ids, skip_special_tokens=True) preds = self.tokenizer.batch_decode(output_ids, skip_special_tokens=True)
preds = [pred.strip() for pred in preds] preds = [pred.strip() for pred in preds]
preds = [{"label": preds[0], "score": 1.0}]
return preds return preds