Test with init_image parameter
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def infer(prompt, refer_image, height, width):
|
|
| 32 |
# num_inference_steps=num_inference_steps,
|
| 33 |
width=width,
|
| 34 |
height=height,
|
| 35 |
-
|
| 36 |
# generator=generator
|
| 37 |
).images[0]
|
| 38 |
|
|
|
|
| 32 |
# num_inference_steps=num_inference_steps,
|
| 33 |
width=width,
|
| 34 |
height=height,
|
| 35 |
+
init_image=refer_image
|
| 36 |
# generator=generator
|
| 37 |
).images[0]
|
| 38 |
|