Update app.py

app.py

@@ -45,9 +45,9 @@ def image_create(seed: int, truncation_psi: float):
             #gaussian=gaussian_radius,
             log_visual_freq=1000
     ))
-    """device = th.device()
+    device = th.device()
     generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
-    latent = torch.randn((1, 512), device=device) 
+    """latent = torch.randn((1, 512), device=device) 
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
     return imgs_arr[0][0]"""