from typing import List

from pydantic import BaseModel


class StableDiffusionResponse(BaseModel):
    """
    Stable diffusion response model

    Attributes:
        images (List[str]): List of JPEG image as base64 encoded
        latency (float): Latency in seconds
    """

    images: List[str]
    latency: float