gchhablani commited on
Commit
f15eef4
1 Parent(s): 2c8f495

Change init example

Browse files
Files changed (1) hide show
  1. apps/mlm.py +1 -1
apps/mlm.py CHANGED
@@ -42,7 +42,7 @@ def app(state):
42
  mlm_checkpoints = ["flax-community/clip-vision-bert-cc12m-70k"]
43
  dummy_data = pd.read_csv("cc12m_data/vqa_val.tsv", sep="\t")
44
 
45
- first_index = 20
46
  # Init Session mlm_state
47
  if mlm_state.mlm_image_file is None:
48
  mlm_state.mlm_image_file = dummy_data.loc[first_index, "image_file"]
 
42
  mlm_checkpoints = ["flax-community/clip-vision-bert-cc12m-70k"]
43
  dummy_data = pd.read_csv("cc12m_data/vqa_val.tsv", sep="\t")
44
 
45
+ first_index = 15
46
  # Init Session mlm_state
47
  if mlm_state.mlm_image_file is None:
48
  mlm_state.mlm_image_file = dummy_data.loc[first_index, "image_file"]