one more try

handler.py

@@ -51,7 +51,7 @@ class EndpointHandler:
         ).images
         image = images[0]
 
-        return {"image": image}
+        return image
 
 
 if __name__ == "__main__":