Huong commited on
Commit
933e631
·
1 Parent(s): 001ffb6

adding code to retrieve HF_TOKEN

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -58,6 +58,7 @@ def download_olmoasr_checkpoint():
58
  filename=CHECKPOINT_FILENAME,
59
  local_dir=LOCAL_CHECKPOINT_DIR,
60
  local_dir_use_symlinks=False,
 
61
  )
62
  print(f"Downloaded checkpoint to {downloaded_path}")
63
  return downloaded_path
 
58
  filename=CHECKPOINT_FILENAME,
59
  local_dir=LOCAL_CHECKPOINT_DIR,
60
  local_dir_use_symlinks=False,
61
+ token=os.getenv("HF_TOKEN"),
62
  )
63
  print(f"Downloaded checkpoint to {downloaded_path}")
64
  return downloaded_path