Commit cae334c4 authored by Morgan Funtowicz's avatar Morgan Funtowicz
Browse files

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

Remove hardcoded mask_token and use the value provided by the tokenizer.
parent 4b1970bb
...@@ -294,7 +294,7 @@ ...@@ -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)"
] ]
}, },
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment