Update app.py

app.py

@@ -5,8 +5,12 @@ import torch
 import pickle
 import types
 
-with open('./network.pkl', 'rb') as pickle_file:
-    _G, _D, G = pickle.load(pickle_file)
+from huggingface_hub import hf_hub_url, cached_download
+
+TOKEN = os.environ['TOKEN']
+
+with open(cached_download(hf_hub_url('i72sijia/ganbanales', 'ganbanales.pkl'), use_auth_token=TOKEN), 'rb') as f:
+    G = pickle.load(f)['G_ema']# torch.nn.Module
     
 device = torch.device("cpu")