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 |
|