myshop-capsule
/

image-super-resolution

Inference Endpoints

Model card Files Files and versions Community

justin-shopcapsule commited on Aug 30, 2023

Commit

07a9231

•

1 Parent(s): 33324d6

Update handler.py

Files changed (1) hide show

handler.py +3 -0

handler.py CHANGED Viewed

@@ -685,6 +685,9 @@ class EndpointHandler():
             # forward pass
             output_image = self.model.predict([image])
             # base64 encode output
             buffered = BytesIO()
             output_image = output_image.convert('RGB')

             # forward pass
             output_image = self.model.predict([image])
+            if isinstance(output_image, list):
+                output_image = output_image[0]
             # base64 encode output
             buffered = BytesIO()
             output_image = output_image.convert('RGB')