Update app.py
Browse files
app.py
CHANGED
@@ -116,7 +116,7 @@ def process(
|
|
116 |
image=resized_image,
|
117 |
width=width,
|
118 |
height=height,
|
119 |
-
num_images_per_prompt =
|
120 |
strength=strength_slider,
|
121 |
generator=generator,
|
122 |
joint_attention_kwargs={"scale": num_influence},
|
@@ -125,7 +125,7 @@ def process(
|
|
125 |
)
|
126 |
print('INFERENCE DONE')
|
127 |
#return result.images[0].resize((image.size), Image.LANCZOS), result.images[1].resize((image.size), Image.LANCZOS)
|
128 |
-
return result.images[0], result.images[1]
|
129 |
|
130 |
#resized_mask = mask.resize((width, height), Image.LANCZOS)
|
131 |
|
|
|
116 |
image=resized_image,
|
117 |
width=width,
|
118 |
height=height,
|
119 |
+
num_images_per_prompt =1,
|
120 |
strength=strength_slider,
|
121 |
generator=generator,
|
122 |
joint_attention_kwargs={"scale": num_influence},
|
|
|
125 |
)
|
126 |
print('INFERENCE DONE')
|
127 |
#return result.images[0].resize((image.size), Image.LANCZOS), result.images[1].resize((image.size), Image.LANCZOS)
|
128 |
+
return result.images[0], None #result.images[1]
|
129 |
|
130 |
#resized_mask = mask.resize((width, height), Image.LANCZOS)
|
131 |
|