Spaces:

feng2022
/

Time-TravelRephotography

Runtime error

feng2022 commited on Sep 7, 2022

Commit

a97ac53

1 Parent(s): 29cc422

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def image_create(seed: int, truncation_psi: float):
     ))
     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","feng2022/Time_TravelRephotography_checkpoint_b",args,device)
     latent = torch.randn((1, 512), device=device)
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)

     ))
     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)