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

Remove hardcoded mask_token and use the value provided by the tokenizer.
2020-03-18 17:11:42 +01:00
parent 4b1970bb4c
commit cae334c43c
1 changed files with 1 additions and 1 deletions
--- a/notebooks/03-pipelines.ipynb
+++ b/notebooks/03-pipelines.ipynb
@@ -294,7 +294,7 @@
   ],
   "source": [
    "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)"
   ]
  },
  {