Spaces:
Running on Zero
Running on Zero
File size: 524 Bytes
9e3b8ca | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | from dataclasses import dataclass
from typing import List, Union
import numpy as np
import PIL.Image
from ...utils import BaseOutput
@dataclass
class JoyImageEditPipelineOutput(BaseOutput):
"""
Output class for JoyImageEdit generation pipelines.
"""
images: Union[List[PIL.Image.Image], np.ndarray]
@dataclass
class JoyImageEditPlusPipelineOutput(BaseOutput):
"""
Output class for JoyImage Edit Plus multi-image editing pipelines.
"""
images: Union[List[PIL.Image.Image], np.ndarray]
|