Spaces:
Running
Running
gavinyuan
commited on
Commit
•
b4cdcef
1
Parent(s):
df7d8d2
update: GPEN weights
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ if fs_model_name == 'faceshifter':
|
|
92 |
if post:
|
93 |
target_hair_mask = trick.get_any_mask(i_t, par=[0, 17])
|
94 |
target_hair_mask = trick.smooth_mask(target_hair_mask)
|
95 |
-
i_r = target_hair_mask * i_t + (target_hair_mask * (-1.) + 1.) * i_r # torch 1.12.
|
96 |
i_r = trick.finetune_mouth(i_s, i_t, i_r) if in_size == 256 else i_r
|
97 |
|
98 |
img_r = trick.tensor_to_arr(i_r)[0]
|
|
|
92 |
if post:
|
93 |
target_hair_mask = trick.get_any_mask(i_t, par=[0, 17])
|
94 |
target_hair_mask = trick.smooth_mask(target_hair_mask)
|
95 |
+
i_r = target_hair_mask * i_t + (target_hair_mask * (-1.) + 1.) * i_r # torch 1.12.0
|
96 |
i_r = trick.finetune_mouth(i_s, i_t, i_r) if in_size == 256 else i_r
|
97 |
|
98 |
img_r = trick.tensor_to_arr(i_r)[0]
|