Update app.py

app.py

@@ -9,9 +9,9 @@ from huggingface_hub import hf_hub_url, cached_download
 
 TOKEN = os.environ['TOKEN']
 
-with open(cached_download('network.pkl', 'rb') as f:
-    G = pickle.load(f)['G_ema']# torch.nn.Module
-
+with open(pkl_file, 'rb') as pickle_file:
+    _G, _D, G = pickle.load(pickle_file)
+    
 device = torch.device("cpu")
 
 if torch.cuda.is_available():