pip install -q transformers datasets
We encourage you to share your model with the community. Log in to your Hugging Face account to upload it to the 🤗 Hub.
When prompted, enter your token to log in:

from huggingface_hub import notebook_login
notebook_login()

Let's define the model checkpoint as a global variable.

model_checkpoint = "dandelin/vilt-b32-mlm"