from pydantic import BaseModel class PromptGenerationImageModel(BaseModel): text: str