Spaces:

ygtxr1997
/

ReliableSwap_Demo

Running

gavinyuan commited on Jun 6, 2023

Commit

0958530

•

1 Parent(s): 5089b6a

update: app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -358,7 +358,7 @@ def swap_image_gr(img1, img2, use_post=False, use_gpen=False, ):
         data_dir,
         T,
         fs_model,
-        gpu_mode=gpu_mode,
         align_target='ffhq',
         align_source='ffhq',
         use_post=use_post,

         data_dir,
         T,
         fs_model,
+        gpu_mode=use_gpu,
         align_target='ffhq',
         align_source='ffhq',
         use_post=use_post,