Update handler.py

handler.py

@@ -19,7 +19,9 @@ class EndpointHandler():
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         image = data.pop("image", None)
-        low_res_img = image.resize((128, 128))
+        # process image
+        image = self.decode_base64_image(image)
+        low_res_img = image#.resize((128, 128))
         
         with torch.no_grad():
             upscaled_image = self.pipeline(low_res_img, num_inference_steps=100, eta=1).images[0]