CorvaeOboro commited on
Commit
815e124
β€’
1 Parent(s): 589ceac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,9 +7,9 @@ import types
7
 
8
  from huggingface_hub import hf_hub_url, cached_download
9
 
10
- TOKEN = os.environ['TOKEN']
11
 
12
- with open(cached_download(hf_hub_url('CorvaeOboro/gen_ability_icon', 'gen_ability_icon_stylegan2ada_20220801.pkl'), use_auth_token=TOKEN), 'rb') as f:
13
  G = pickle.load(f)['G_ema']# torch.nn.Module
14
 
15
  device = torch.device("cpu")
 
7
 
8
  from huggingface_hub import hf_hub_url, cached_download
9
 
10
+ #TOKEN = os.environ['TOKEN']
11
 
12
+ with open(cached_download(hf_hub_url('CorvaeOboro/gen_ability_icon', 'gen_ability_icon_stylegan2ada_20220801.pkl')), 'rb') as f:
13
  G = pickle.load(f)['G_ema']# torch.nn.Module
14
 
15
  device = torch.device("cpu")