feng2022 commited on
Commit
ed89585
1 Parent(s): 9d9b9ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def image_create(input_img):
60
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
61
  #gaussian=gaussian_radius,
62
  log_visual_freq=1000
63
- )
64
  device = th.device()
65
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
66
  latent = torch.randn((1, 512), device=device)
 
60
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
61
  #gaussian=gaussian_radius,
62
  log_visual_freq=1000
63
+ ))
64
  device = th.device()
65
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
66
  latent = torch.randn((1, 512), device=device)