feng2022 commited on
Commit
7b5488e
1 Parent(s): c37dd70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -79,7 +79,7 @@ def main():
79
  # ini = "False1"
80
  #result = subprocess.check_output(['nvidia-smi'])
81
  #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)
82
- """args = ProjectorArguments().parse(
83
  args=[str(input_path)],
84
  namespace=Namespace(
85
  # spectral_sensitivity=spectral_sensitivity,
@@ -93,7 +93,7 @@ def main():
93
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
94
  latent = torch.randn((1, 512), device=device)
95
  img_out, _, _ = generator([latent])
96
- imgs_arr = make_image(img_out)"""
97
  #iface = gr.Interface(
98
  #fn=predict,
99
  #inputs='text',
 
79
  # ini = "False1"
80
  #result = subprocess.check_output(['nvidia-smi'])
81
  #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)
82
+ args = ProjectorArguments().parse(
83
  args=[str(input_path)],
84
  namespace=Namespace(
85
  # spectral_sensitivity=spectral_sensitivity,
 
93
  generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
94
  latent = torch.randn((1, 512), device=device)
95
  img_out, _, _ = generator([latent])
96
+ imgs_arr = make_image(img_out)
97
  #iface = gr.Interface(
98
  #fn=predict,
99
  #inputs='text',