Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -43,8 +43,8 @@ def image_create(seed: int, truncation_psi: float):
|
|
43 |
log_visual_freq=1000
|
44 |
))
|
45 |
device = th.device()
|
46 |
-
|
47 |
-
generator = create_generator("checkpoint_b.pt.pth","feng2022/Time_TravelRephotography_checkpoint_b",args,device)
|
48 |
latent = torch.randn((1, 512), device=device)
|
49 |
img_out, _, _ = generator([latent])
|
50 |
imgs_arr = make_image(img_out)
|
|
|
43 |
log_visual_freq=1000
|
44 |
))
|
45 |
device = th.device()
|
46 |
+
generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
|
47 |
+
#generator = create_generator("checkpoint_b.pt.pth","feng2022/Time_TravelRephotography_checkpoint_b",args,device)
|
48 |
latent = torch.randn((1, 512), device=device)
|
49 |
img_out, _, _ = generator([latent])
|
50 |
imgs_arr = make_image(img_out)
|