Update handler.py

handler.py

@@ -53,6 +53,13 @@ class EndpointHandler():
         
         prompt = data.pop("prompt", "")
 
+        if(method == "smooth"):
+            if encoded_image is not None:
+                image = self.decode_base64_image(encoded_image)
+                out = self.smooth_pipe(prompt, image=image).images[0]
+    
+                return out
+
         negative_prompt = data.pop("negative_prompt", "")
 
         method = data.pop("method", "slow")
@@ -82,11 +89,6 @@ class EndpointHandler():
     
             return image
 
-        if(method == "smooth"):
-            image = self.smooth_pipe(prompt, image=image).images
-
-            return image
-
         #pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
 
         self.pipe.enable_xformers_memory_efficient_attention()