Spaces:

feng2022
/

Time-TravelRephotography

Runtime error

feng2022 commited on Sep 7, 2022

Commit

b74c99a

1 Parent(s): a97ac53

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

             log_visual_freq=1000
     ))
     device = th.device()
+    generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
+    #generator = create_generator("checkpoint_b.pt.pth","feng2022/Time_TravelRephotography_checkpoint_b",args,device)
     latent = torch.randn((1, 512), device=device)
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)