saassa
/

newha

saassa commited on Sep 24, 2023

Commit

b036aaf

1 Parent(s): c9b58ae

Update handler.py

Files changed (1) hide show

handler.py CHANGED Viewed

@@ -133,7 +133,7 @@ class EndpointHandler():
         # generator=self.generator COMMENTED from self.pipe
         # return the first generated PIL image
-        return out.images[0]
     # helper to decode input image
     def decode_base64_image(self, image_string):

         # generator=self.generator COMMENTED from self.pipe
         # return the first generated PIL image
+        return out.images
     # helper to decode input image
     def decode_base64_image(self, image_string):