Update app.py

app.py

@@ -6,9 +6,12 @@ import pickle
 import types
 
 from huggingface_hub import hf_hub_url, cached_download
+from huggingface_hub import hf_hub_download
+
+#hf_hub_download(repo_id="CorvaeOboro/gen_ability_icon", filename="gen_ability_icon_stylegan2ada_20221012.pkl", repo_type="dataset")
 
 #TOKEN = os.environ['TOKEN']
-with open(cached_download(hf_hub_url('CorvaeOboro/gen_ability_icon', 'gen_ability_icon_stylegan2ada_20221012.pkl')), 'rb') as f:
+with open(hf_hub_download(repo_id="CorvaeOboro/gen_ability_icon", filename="gen_ability_icon_stylegan2ada_20221012.pkl", repo_type="dataset"), 'rb') as f:
     G = pickle.load(f)['G_ema']# torch.nn.Module
 
 device = torch.device("cpu")