fffiloni commited on
Commit
4efa9a6
1 Parent(s): 86cbf7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,7 +1,11 @@
1
  import gradio as gr
 
2
  import subprocess
3
  from huggingface_hub import snapshot_download
4
 
 
 
 
5
  def set_accelerate_default_config():
6
  try:
7
  subprocess.run(["accelerate", "config", "default"], check=True)
@@ -60,6 +64,7 @@ def main(dataset_url):
60
  local_dir=local_dir,
61
  repo_type="dataset",
62
  ignore_patterns=".gitattributes",
 
63
  )
64
 
65
  set_accelerate_default_config()
 
1
  import gradio as gr
2
+ import os
3
  import subprocess
4
  from huggingface_hub import snapshot_download
5
 
6
+ hf_token = os.environ.get["HF_TOKEN"]
7
+ print(hf_token)
8
+
9
  def set_accelerate_default_config():
10
  try:
11
  subprocess.run(["accelerate", "config", "default"], check=True)
 
64
  local_dir=local_dir,
65
  repo_type="dataset",
66
  ignore_patterns=".gitattributes",
67
+ token=hf_token
68
  )
69
 
70
  set_accelerate_default_config()