Update app.py

app.py

@@ -68,7 +68,10 @@ def make_transform(translate: Tuple[float,float], angle: float):
 
 #----------------------------------------------------------------------------
 
-
+device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
+with dnnlib.util.open_url('https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/pokemon.pkl') as f:
+    G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore
+    
 def generate_images(seeds):
     """Generate images using pretrained network pickle.
     Examples:
@@ -82,9 +85,7 @@ def generate_images(seeds):
         --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-t-metfacesu-1024x1024.pkl
     """
 
-    device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
-    with dnnlib.util.open_url('https://s3.eu-central-1.amazonaws.com/avg-projects/projected_gan/models/pokemon.pkl') as f:
-        G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore
+
 
 
     # Labels.