Update app.py

app.py

@@ -115,7 +115,9 @@ model_name = "stylegan3-t-ffhqu-1024x1024.pkl"
 #model_name = "stylegan3-t-afhqv2-512x512.pkl"
 network_url = base_url + model_name
 
-with open(fetch_model(network_url), 'rb') as fp:
+os.system("wget -c https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-t-ffhqu-1024x1024.pkl")
+
+with open('stylegan3-t-ffhqu-1024x1024.pkl', 'rb') as fp:
   G = pickle.load(fp)['G_ema'].to(device)
 
 zs = torch.randn([10000, G.mapping.z_dim], device=device)