Update handler.py
Browse files- handler.py +2 -0
handler.py
CHANGED
@@ -135,6 +135,8 @@ def generate_image_with_conditioning_scale(**inputs):
|
|
135 |
|
136 |
def generate_image(pipe, inputs):
|
137 |
styles = inputs["styles"]
|
|
|
|
|
138 |
content = inputs["content"]
|
139 |
art_scale = inputs["art_scale"]
|
140 |
|
|
|
135 |
|
136 |
def generate_image(pipe, inputs):
|
137 |
styles = inputs["styles"]
|
138 |
+
if isinstance(styles, str):
|
139 |
+
styles = [styles]
|
140 |
content = inputs["content"]
|
141 |
art_scale = inputs["art_scale"]
|
142 |
|