feng2022 commited on
Commit
fee5386
1 Parent(s): 0f60bc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -91,8 +91,8 @@ def main():
91
  device = th.device()
92
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
93
  latent = torch.randn((1, 512), device=device)
94
- img_out, _, _ = generator([latent])
95
- imgs_arr = make_image(img_out)
96
  iface = gr.Interface(
97
  fn=predict,
98
  inputs='text',
 
91
  device = th.device()
92
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
93
  latent = torch.randn((1, 512), device=device)
94
+ #img_out, _, _ = generator([latent])
95
+ #imgs_arr = make_image(img_out)
96
  iface = gr.Interface(
97
  fn=predict,
98
  inputs='text',