Spaces:
Sleeping
Sleeping
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
|