Spaces:

feng2022
/

Time-TravelRephotography

Runtime error

feng2022 commited on Sep 1, 2022

Commit

1409c53

1 Parent(s): 0dce906

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,8 +45,8 @@ def image_create(seed: int, truncation_psi: float):
             #gaussian=gaussian_radius,
             log_visual_freq=1000
     ))
-    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)
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)

             #gaussian=gaussian_radius,
             log_visual_freq=1000
     ))
+    #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)
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)