Spaces:

CorvaeOboro
/

gen_ability_icon

Runtime error

CorvaeOboro commited on Aug 19, 2022

Commit

b5bbbf0

•

1 Parent(s): e8f045f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from huggingface_hub import hf_hub_url, cached_download
 #TOKEN = os.environ['TOKEN']
 with open(cached_download(hf_hub_url('CorvaeOboro/gen_ability_icon', 'gen_ability_icon_stylegan2ada_20220801.pkl')), 'rb') as f:
-    G = pickle.load(f)['G_ema']# torch.nn.Module
 device = torch.device("cpu")
 if torch.cuda.is_available():

 #TOKEN = os.environ['TOKEN']
 with open(cached_download(hf_hub_url('CorvaeOboro/gen_ability_icon', 'gen_ability_icon_stylegan2ada_20220801.pkl')), 'rb') as f:
+    G = pickle.load(f)['f']# torch.nn.Module
 device = torch.device("cpu")
 if torch.cuda.is_available():