update

app.py

@@ -113,7 +113,7 @@ def image_generation(input_image, quality, use_target_pose, pose, dimension, pro
         else:
             _, _, images, *_ = generator(features)
 
-    images = ((images.permute(0, 2, 3, 1).detach().cpu().numpy() + 1) / 2 * 255).astype(np.uint8)
+    images = ((images.permute(0, 2, 3, 1).clip(-1, 1).detach().cpu().numpy() + 1) / 2 * 255).astype(np.uint8)
     for image in progress.tqdm(images, desc="Processing images"):
         generated_images.append(Image.fromarray(image))