Spaces:
Running
Running
gavinyuan
commited on
Commit
·
396310b
1
Parent(s):
b4cdcef
update: app.py
Browse files
app.py
CHANGED
|
@@ -317,7 +317,7 @@ def process_video(
|
|
| 317 |
output = G(source_img, target)
|
| 318 |
target_hair_mask = trick.get_any_mask(target, par=[0, 17])
|
| 319 |
target_hair_mask = trick.smooth_mask(target_hair_mask)
|
| 320 |
-
output = target_hair_mask * target + (target_hair_mask * (-1) + 1) * output
|
| 321 |
output = trick.finetune_mouth(source_img, target, output)
|
| 322 |
elif 'simswap' in fs_model_name and 'official' not in fs_model_name:
|
| 323 |
output = fs_model(source=source_img, target=target,
|
|
@@ -325,7 +325,7 @@ def process_video(
|
|
| 325 |
if 'vanilla' not in fs_model_name:
|
| 326 |
target_hair_mask = trick.get_any_mask(target, par=[0, 17])
|
| 327 |
target_hair_mask = trick.smooth_mask(target_hair_mask)
|
| 328 |
-
output = target_hair_mask * target + (target_hair_mask * (-1) + 1) * output
|
| 329 |
output = trick.finetune_mouth(source_img, target, output)
|
| 330 |
output = output.clamp(-1, 1)
|
| 331 |
elif 'simswap_official' in fs_model_name:
|
|
|
|
| 317 |
output = G(source_img, target)
|
| 318 |
target_hair_mask = trick.get_any_mask(target, par=[0, 17])
|
| 319 |
target_hair_mask = trick.smooth_mask(target_hair_mask)
|
| 320 |
+
output = target_hair_mask * target + (target_hair_mask * (-1.) + 1.) * output
|
| 321 |
output = trick.finetune_mouth(source_img, target, output)
|
| 322 |
elif 'simswap' in fs_model_name and 'official' not in fs_model_name:
|
| 323 |
output = fs_model(source=source_img, target=target,
|
|
|
|
| 325 |
if 'vanilla' not in fs_model_name:
|
| 326 |
target_hair_mask = trick.get_any_mask(target, par=[0, 17])
|
| 327 |
target_hair_mask = trick.smooth_mask(target_hair_mask)
|
| 328 |
+
output = target_hair_mask * target + (target_hair_mask * (-1.) + 1.) * output
|
| 329 |
output = trick.finetune_mouth(source_img, target, output)
|
| 330 |
output = output.clamp(-1, 1)
|
| 331 |
elif 'simswap_official' in fs_model_name:
|