Update app.py

app.py

@@ -25,6 +25,7 @@ from projector import (
 )
 sys.path.insert(0, 'StyleGAN-Human')
 
+spectral_sensitivity =  'b'
 TITLE = 'Time-TravelRephotography'
 DESCRIPTION = '''This is an unofficial demo for https://github.com/Time-Travel-Rephotography.
 '''
@@ -73,7 +74,15 @@ def main():
     #    ini = "False1"
     #result = subprocess.check_output(['nvidia-smi'])
     
-    #args = parse_args()
+    args = ProjectorArguments().parse(
+        args=[str(input_path)], 
+        namespace=Namespace(
+            # spectral_sensitivity=spectral_sensitivity,
+            encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
+            # encoder_name=spectral_sensitivity,
+            # gaussian=gaussian_radius,
+            log_visual_freq=1000,
+))
     #device = torch.device(args.device)
     #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)