Update app.py

app.py

@@ -11,7 +11,7 @@ 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(hf_hub_download(repo_id="CorvaeOboro/gen_ability_icon", filename="gen_ability_icon_stylegan2ada_20221012.pkl", repo_type="dataset"), 'rb') as f:
+with open(hf_hub_download(repo_id="CorvaeOboro/gen_ability_icon", filename="gen_ability_icon_stylegan2ada_20221012.pkl", repo_type="model"), 'rb') as f:
     G = pickle.load(f)['G_ema']# torch.nn.Module
 
 device = torch.device("cpu")