gavinyuan commited on
Commit
b4cdcef
1 Parent(s): df7d8d2

update: GPEN weights

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]