MatteoScript commited on
Commit
b134cae
·
verified ·
1 Parent(s): d57ded5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -6
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: Enum = Enum("StyleEnum", [(style, style) for style in style_image.keys()])
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