From 593df264183ea2d145ecc95507b7706ecaa7e897 Mon Sep 17 00:00:00 2001
From: Yiheng <yihengmen@gmail.com>
Date: Thu, 30 May 2024 15:06:02 +0800
Subject: [PATCH] feat: return success

---
 image_c_postprocessor.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/image_c_postprocessor.py b/image_c_postprocessor.py
index 9bbf47f..88dbbfc 100644
--- a/image_c_postprocessor.py
+++ b/image_c_postprocessor.py
@@ -112,7 +112,7 @@ class ImageCPostprocessor:
                 pil_image = tensor2pil(img)
 
                 # Apply Fliter
-                new_img = self.process_image_pil(pil_image, kernel_size)
+                new_img, success = self.process_image_pil(pil_image, kernel_size)
 
                 # Output image
                 out_image = (pil2tensor(new_img) if pil_image else img)
@@ -129,14 +129,14 @@ class ImageCPostprocessor:
             pil_image = tensor2pil(img)
 
             # Apply Fliter
-            new_img = self.process_image_pil(pil_image, kernel_size)
+            new_img, success = self.process_image_pil(pil_image, kernel_size)
 
             # Output image
             out_image = (pil2tensor(new_img) if pil_image else img)
 
             tensors = out_image
 
-        return (tensors, )
+        return (tensors, success)
 
 class PrivateImageMask:
     def __init__(self):