from pydantic import BaseModel


class PromptGenerationImageModel(BaseModel):
    text: str


class ImageRequest(BaseModel):
    img: str