app/models/meme.py

from pydantic import BaseModel


class Meme(BaseModel):
    caption: str
    image_url: str

class CaptionRequest(BaseModel):
    prompt: str
    max_length: int = 50
    num_return_sequences: int = 3