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