Update handler.py

handler.py

@@ -9,8 +9,9 @@ class EndpointHandler():
         self.upscaler = UpscaleWithModel.from_pretrained("DAT_x3.pth").to("cuda")
         
     def __call__(self, data):
-        inputs = data.pop("inputs", data)
-        output = upscaler(inputs, tiling=True, tile_width=768, tile_height=768, overlap=8)
+        img = data.pop("inputs", data)
+        img = Image.open(BytesIO(img))
+        output = upscaler(img, tiling=True, tile_width=768, tile_height=768, overlap=8)
         
         output = outputs.reconstruction.data.squeeze().float().cpu().clamp_(0, 1).numpy()
         output = np.moveaxis(output, source=0, destination=-1)