sections/mlm_usage.md

• This demo loads the FlaxCLIPVisionBertForMaskedLM present in the model directory of this repository. The checkpoint is loaded from flax-community/clip-vision-bert-cc12m-70k which is pre-trained checkpoint with 70k steps. 100 random validation set examples are present in the cc12m_data/vqa_val.tsv with respective images in the cc12m_data/images_data directory.

• We provide English Translation of the caption for users who are not well-acquainted with the other languages. This is done using mtranslate to keep things flexible enough and needs internet connection as it uses the Google Translate API.

• The model predicts the scores for tokens from the bert-base-multilingual-uncased checkpoint.

• The top-5 predictions are displayed below and their respective confidence scores are shown in form of a bar plot.