CorvaeOboro commited on
Commit
85d7c39
1 Parent(s): 8a75545

Update app.py

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