Update handler.py

add count back to output

handler.py

@@ -123,7 +123,7 @@ class EndpointHandler():
 
         
         # return first generate PIL image
-        return out.images[0]
+        return out.images[0, count]
     
     # helper to decode input image
     def decode_base64_image(self, image_string):