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