File size: 286 Bytes
29184ba
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from abc import ABC, abstractmethod
from typing import List

class ImageProvider(ABC):
    @abstractmethod
    def generate_image(self, prompt: str, size: str = "1024x1024", n: int = 1) -> List[str]:
        """Return a list of base64-encoded PNGs."""
        raise NotImplementedError