VictorSanh commited on
Commit
6893a8c
β€’
1 Parent(s): 7ea3311
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ else:
42
  ) ** 2
43
  BOS_TOKEN = PROCESSOR.tokenizer.bos_token
44
  BAD_WORDS_IDS = PROCESSOR.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
45
- DATASET = load_dataset("HuggingFaceM4/RAVEN_rendered", split="validation")
46
 
47
  ## Utils
48
 
 
42
  ) ** 2
43
  BOS_TOKEN = PROCESSOR.tokenizer.bos_token
44
  BAD_WORDS_IDS = PROCESSOR.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
45
+ DATASET = load_dataset("HuggingFaceM4/RAVEN_rendered", split="validation", token=os.environ["HF_AUTH_TOKEN"])
46
 
47
  ## Utils
48