13 lines
460 B
Python
13 lines
460 B
Python
from __future__ import annotations
|
|
|
|
from . import response_helper
|
|
|
|
def handle_response(api_result, schema=None, loaded_client=None, response_url=None):
|
|
generated_images = getattr(api_result, "generated_images", None)
|
|
if not generated_images:
|
|
return [None, None]
|
|
|
|
image = getattr(generated_images[0], "image", None)
|
|
image_bytes = getattr(image, "image_bytes", None)
|
|
return [response_helper.bytes_to_tensor(image_bytes), image_bytes]
|