Update app.py

app.py

@@ -72,9 +72,15 @@ def make_transform(translate: Tuple[float,float], angle: float):
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
 config_file_url = hf_hub_url("autonomousvision/Projected_GAN_Pokemon", filename="pokemon.pkl")
+
 with dnnlib.util.open_url(config_file_url) as f:
     G = legacy.load_network_pkl(f)['G_ema'].to(device)  # type: ignore
 
+for m in G.modules():
+    if isinstance(m, torch.nn.Upsample):
+        if not hasattr(m, 'recompute_scale_factor'):
+            m.recompute_scale_factor = None
+
 
 def generate_images(seeds):
     label = torch.zeros([1, G.c_dim], device=device)