feng2022 commited on
Commit
b076c7e
1 Parent(s): b0dd76b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def image_create(input_img):
64
  log_visual_freq=1000,
65
  input='text',
66
  )
67
- generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
68
  latent = torch.randn((1, 512), device=device)
69
  img_out, _, _ = generator([latent])
70
  imgs_arr = make_image(img_out)
 
64
  log_visual_freq=1000,
65
  input='text',
66
  )
67
+ generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args)
68
  latent = torch.randn((1, 512), device=device)
69
  img_out, _, _ = generator([latent])
70
  imgs_arr = make_image(img_out)