helloworld-S commited on
Commit
97f40ec
·
verified ·
1 Parent(s): 08958e6

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -31,6 +31,8 @@ import shutil
31
  import yaml
32
  import numpy as np
33
  from huggingface_hub import hf_hub_download
 
 
34
 
35
  dtype = torch.bfloat16
36
  device = "cuda"
 
31
  import yaml
32
  import numpy as np
33
  from huggingface_hub import hf_hub_download
34
+ import huggingface_hub
35
+ huggingface_hub.login(os.getenv('HF_TOKEN'))
36
 
37
  dtype = torch.bfloat16
38
  device = "cuda"