Spaces:

flax-community
/

Multilingual-VQA

Runtime error

gchhablani commited on Jul 24, 2021

Commit

875bee5

•

1 Parent(s): 63672a5

Fix typo

Files changed (1) hide show

apps/mlm.py CHANGED Viewed

@@ -53,7 +53,7 @@ def app(state):
         mlm_state.unmasked_caption = caption
         ids = bert_tokenizer.encode(caption)
         mask_index = np.random.randint(1, len(ids) - 1)
-        mlm_state.currently_masked_token = bert_tokenizer.convert_ids_to_tokens[[ids[mask_index]]][0]
         ids[mask_index] = bert_tokenizer.mask_token_id
         mlm_state.caption = bert_tokenizer.decode(ids[1:-1])
         mlm_state.caption_lang_id = dummy_data.loc[first_index, "lang_id"]
@@ -83,7 +83,7 @@ def app(state):
         mlm_state.unmasked_caption = caption
         ids = bert_tokenizer.encode(caption)
         mask_index = np.random.randint(1, len(ids) - 1)
-        mlm_state.currently_masked_token = bert_tokenizer.convert_ids_to_tokens[[ids[mask_index]]][0]
         ids[mask_index] = bert_tokenizer.mask_token_id
         mlm_state.caption = bert_tokenizer.decode(ids[1:-1])
         mlm_state.caption_lang_id = sample.loc[0, "lang_id"]

         mlm_state.unmasked_caption = caption
         ids = bert_tokenizer.encode(caption)
         mask_index = np.random.randint(1, len(ids) - 1)
+        mlm_state.currently_masked_token = bert_tokenizer.convert_ids_to_tokens([ids[mask_index]])[0]
         ids[mask_index] = bert_tokenizer.mask_token_id
         mlm_state.caption = bert_tokenizer.decode(ids[1:-1])
         mlm_state.caption_lang_id = dummy_data.loc[first_index, "lang_id"]
         mlm_state.unmasked_caption = caption
         ids = bert_tokenizer.encode(caption)
         mask_index = np.random.randint(1, len(ids) - 1)
+        mlm_state.currently_masked_token = bert_tokenizer.convert_ids_to_tokens([ids[mask_index]])[0]
         ids[mask_index] = bert_tokenizer.mask_token_id
         mlm_state.caption = bert_tokenizer.decode(ids[1:-1])
         mlm_state.caption_lang_id = sample.loc[0, "lang_id"]