SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Improve fill-mask pipeline example in 03-pipelines notebook.

Browse Source 
Remove hardcoded mask_token and use the value provided by the tokenizer.
This commit is contained in:
Morgan Funtowicz
2020-03-18 17:11:42 +01:00
parent 4b1970bb4c
commit cae334c43c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
notebooks/03-pipelines.ipynb
View File

@@ -294,7 +294,7 @@
], ],
"source": [ "source": [
"nlp_fill = pipeline('fill-mask')\n", "nlp_fill = pipeline('fill-mask')\n",
"nlp_fill('Hugging Face is a French company based in <mask>')" "nlp_fill('Hugging Face is a French company based in ' + nlp_fill.tokenizer.mask_token)"
] ]
}, },
{ {