StableDesign2 / model.py
anbucur
Added functionality
4e4b650
raw
history blame contribute delete
518 Bytes
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")