update: app.py

app.py

@@ -314,10 +314,10 @@ def process_video(
         ''' Face swapping '''
         with torch.no_grad():
             if 'faceshifter' in fs_model_name:
-                output = fs_model(source_img, target)
+                output = fs_model(source_img, target)[0]
                 target_hair_mask = trick.get_any_mask(target, par=[0, 17])
                 target_hair_mask = trick.smooth_mask(target_hair_mask)
-                print(output.shape, target.shape, target_hair_mask.shape)
+                # print(output.shape, target.shape, target_hair_mask.shape)
                 output = target_hair_mask * target + (target_hair_mask * (-1.) + 1.) * output
                 output = trick.finetune_mouth(source_img, target, output)
             elif 'simswap' in fs_model_name and 'official' not in fs_model_name: