Update app.py

app.py

@@ -9,8 +9,8 @@ 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)['f']# torch.nn.Module
+with open(cached_download(hf_hub_url('CorvaeOboro/gen_ability_icon', 'gen_ability_icon_stylegan2ada_20220819.pkl')), 'rb') as f:
+    G = pickle.load(f)['G_ema']# torch.nn.Module
 
 device = torch.device("cpu")
 if torch.cuda.is_available():