saassa
/

newha

saassa commited on Sep 25, 2023

Commit

fdacaf1

•

1 Parent(s): cf6c6b4

Update handler.py

Files changed (1) hide show

handler.py CHANGED Viewed

@@ -132,7 +132,7 @@ class EndpointHandler():
         # 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):

         # 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):