feng2022 commited on
Commit
092a462
1 Parent(s): 5156054

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -26,6 +26,7 @@ from projector import (
26
  )
27
  sys.path.insert(0, 'StyleGAN-Human')
28
 
 
29
  spectral_sensitivity = 'b'
30
  TITLE = 'Time-TravelRephotography'
31
  DESCRIPTION = '''This is an unofficial demo for https://github.com/Time-Travel-Rephotography.
@@ -76,6 +77,7 @@ def main():
76
  #result = subprocess.check_output(['nvidia-smi'])
77
 
78
  args = ProjectorArguments().parse(
 
79
  namespace=Namespace(
80
  # spectral_sensitivity=spectral_sensitivity,
81
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
 
26
  )
27
  sys.path.insert(0, 'StyleGAN-Human')
28
 
29
+ input_path = ''
30
  spectral_sensitivity = 'b'
31
  TITLE = 'Time-TravelRephotography'
32
  DESCRIPTION = '''This is an unofficial demo for https://github.com/Time-Travel-Rephotography.
 
77
  #result = subprocess.check_output(['nvidia-smi'])
78
 
79
  args = ProjectorArguments().parse(
80
+ args=[str(input_path)],
81
  namespace=Namespace(
82
  # spectral_sensitivity=spectral_sensitivity,
83
  encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",