Update app.py

app.py

@@ -64,7 +64,7 @@ def image_create(input_img):
             log_visual_freq=1000,
             input='text',
         )
-    generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args)
+    generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
     latent = torch.randn((1, 512), device=device) 
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
@@ -80,7 +80,7 @@ def main():
     args = parse_args()
     device = th.device()
     func = functools.partial(image_create, device=device)
-    func = functools.update_wrapper(func, generate_image)
+    func = functools.update_wrapper(func, image_create)
     iface = gr.Interface(
           func, 
           [