| from pydantic import BaseModel | |
| from typing import List | |
| from PIL.Image import Image | |
| from .utils import download_image_as_pil | |
| class TextRequest(BaseModel): | |
| texts: List[str] | |
| class ImageRequest(BaseModel): | |
| urls: List[str] | |
| def download(self) -> List[Image]: | |
| return [download_image_as_pil(url) for url in self.urls] | |
| class Response(BaseModel): | |
| embeddings: List[List[float]] | |