Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,6 +106,7 @@ def generate(
|
|
| 106 |
guidance_scale=guidance_scale_base,
|
| 107 |
num_inference_steps=num_inference_steps_base,
|
| 108 |
generator=generator,
|
|
|
|
| 109 |
output_type="pil",
|
| 110 |
).images[0]
|
| 111 |
else:
|
|
|
|
| 106 |
guidance_scale=guidance_scale_base,
|
| 107 |
num_inference_steps=num_inference_steps_base,
|
| 108 |
generator=generator,
|
| 109 |
+
image=init_image,
|
| 110 |
output_type="pil",
|
| 111 |
).images[0]
|
| 112 |
else:
|