feng2022 commited on
Commit
fd26580
1 Parent(s): a9f4e65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,14 +38,14 @@ ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=Time-
38
  TOKEN = "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"
39
 
40
  def image_create(seed: int, truncation_psi: float):
41
- """args = ProjectorArguments().parse(
42
  args=[str(input_path)],
43
  namespace=Namespace(
44
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
45
  #gaussian=gaussian_radius,
46
  log_visual_freq=1000
47
  ))
48
- device = th.device()
49
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
50
  latent = torch.randn((1, 512), device=device)
51
  img_out, _, _ = generator([latent])
 
38
  TOKEN = "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"
39
 
40
  def image_create(seed: int, truncation_psi: float):
41
+ args = ProjectorArguments().parse(
42
  args=[str(input_path)],
43
  namespace=Namespace(
44
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
45
  #gaussian=gaussian_radius,
46
  log_visual_freq=1000
47
  ))
48
+ """device = th.device()
49
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
50
  latent = torch.randn((1, 512), device=device)
51
  img_out, _, _ = generator([latent])