Spaces:
Running
Running
from typing import List | |
from PIL import Image | |
import numpy as np | |
class DesignModel: | |
"""Interface for the design model""" | |
def __init__(self): | |
self.seed = None | |
self.neg_prompt = None | |
self.additional_quality_suffix = None | |
def generate_design(self, image: Image.Image, num_variations: int = 1, **kwargs) -> List[np.ndarray]: | |
"""Generate design variations from the input image""" | |
raise NotImplementedError("This method should be implemented by concrete model classes") |