From 634d1d512aada6a94647d734f0c460dcf0205fdb Mon Sep 17 00:00:00 2001
From: Art Gourieff <85128026+Gourieff@users.noreply.github.com>
Date: Mon, 19 Jun 2023 16:54:18 +0700
Subject: [PATCH] Fixes

---
 scripts/cimage.py  | 4 ----
 scripts/swapper.py | 8 +-------
 2 files changed, 1 insertion(+), 11 deletions(-)

diff --git a/scripts/cimage.py b/scripts/cimage.py
index 6203d09..bd154e8 100644
--- a/scripts/cimage.py
+++ b/scripts/cimage.py
@@ -23,7 +23,3 @@ def check_batch(x: torch.Tensor) -> torch.Tensor:
     x_checked_image = x_samples_ddim_numpy
     x = torch.from_numpy(x_checked_image).permute(0, 3, 1, 2)
     return x
-
-
-def convert_to_sd(img: Image) -> Image:
-    return img
\ No newline at end of file
diff --git a/scripts/swapper.py b/scripts/swapper.py
index aa5b633..f1d8a0b 100644
--- a/scripts/swapper.py
+++ b/scripts/swapper.py
@@ -11,7 +11,6 @@ from PIL import Image
 
 import insightface
 import onnxruntime
-from scripts.cimage import convert_to_sd
 
 from modules.face_restoration import FaceRestoration, restore_faces
 from modules.upscaler import Upscaler, UpscalerData
@@ -29,10 +28,6 @@ class UpscaleOptions:
     restorer_visibility: float = 0.5
 
 
-def save_image(img: Image, filename: str):
-    convert_to_sd(img).save(filename)
-
-
 def cosine_distance(vector1: np.ndarray, vector2: np.ndarray) -> float:
     vec1 = vector1.flatten()
     vec2 = vector2.flatten()
@@ -162,8 +157,7 @@ def swap_face(
             if upscale_options is not None:
                 result_image = upscale_image(result_image, upscale_options)
 
-            # save_image(result_image, fn.name)
         else:
             logger.info("No source face found")
-    save_image(result_image, fn.name)
+    result_image.save(fn.name)
     return ImageResult(path=fn.name)