Haiyu Wu
commited on
Commit
·
7aa5b58
1
Parent(s):
918e8a0
vec2face demo
Browse files
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
|
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:
|