Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -162,15 +162,10 @@ style_image = {
|
|
162 |
}
|
163 |
}
|
164 |
|
165 |
-
def create_style_enum(style_image):
|
166 |
-
return Enum("StyleEnum", [(style, style) for style in style_image.keys()])
|
167 |
-
|
168 |
-
enum_styles = create_style_enum(style_image)
|
169 |
-
|
170 |
class InputImage(BaseModel):
|
171 |
input: str
|
172 |
negativePrompt: str = ''
|
173 |
-
style:
|
174 |
steps: int = 30
|
175 |
cfg: int = 7
|
176 |
seed: int = -1
|
|
|
162 |
}
|
163 |
}
|
164 |
|
|
|
|
|
|
|
|
|
|
|
165 |
class InputImage(BaseModel):
|
166 |
input: str
|
167 |
negativePrompt: str = ''
|
168 |
+
style: str = ''
|
169 |
steps: int = 30
|
170 |
cfg: int = 7
|
171 |
seed: int = -1
|