image-super-resolution

justin-shopcapsule commited on Aug 30, 2023

Commit

33324d6

1 Parent(s): 87d4dc9

Update handler.py

Files changed (1) hide show

handler.py CHANGED Viewed

@@ -683,7 +683,7 @@ class EndpointHandler():
             image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
             # forward pass
-            output_image = self.model.predict(image)
             # base64 encode output
             buffered = BytesIO()

             image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
             # forward pass
+            output_image = self.model.predict([image])
             # base64 encode output
             buffered = BytesIO()