Update handler.py

handler.py

@@ -34,8 +34,9 @@ class EndpointHandler():
 
         if encoded_image is not None:
             image = self.decode_base64_image(encoded_image)
+            print(image.shape)
             self.smooth_pipe.enable_xformers_memory_efficient_attention()
-            out = self.smooth_pipe(prompt, image=image, ).images[0]
+            out = self.smooth_pipe(prompt, image=image).images[0]
 
             return out