Haiyu Wu commited on
Commit
7aa5b58
·
1 Parent(s): 918e8a0

vec2face demo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def image_generation(input_image, quality, use_target_pose, pose, dimension):
92
  else:
93
  features = torch.repeat_interleave(torch.tensor(feature), 6, dim=0)
94
  features = sample_nearby_vectors(features, [0.7], [1]).float()
95
- if quality > 25 and pose > 20:
96
  images, _ = generator.gen_image(features, quality_model, id_model, pose_model=pose_model,
97
  q_target=quality, pose=pose, class_rep=features)
98
  else:
 
92
  else:
93
  features = torch.repeat_interleave(torch.tensor(feature), 6, dim=0)
94
  features = sample_nearby_vectors(features, [0.7], [1]).float()
95
+ if quality > 25 or pose > 20:
96
  images, _ = generator.gen_image(features, quality_model, id_model, pose_model=pose_model,
97
  q_target=quality, pose=pose, class_rep=features)
98
  else: