Spaces:
Running
Running
zhiweili
commited on
Commit
·
7397cc1
1
Parent(s):
13b9583
fix canny image
Browse files- app_haircolor_pix2pix.py +2 -0
app_haircolor_pix2pix.py
CHANGED
|
@@ -78,6 +78,8 @@ def image_to_image(
|
|
| 78 |
prompt=edit_prompt,
|
| 79 |
negative_prompt=DEFAULT_NEGATIVE_PROMPT,
|
| 80 |
image=input_image,
|
|
|
|
|
|
|
| 81 |
guidance_scale=guidance_scale,
|
| 82 |
image_guidance_scale=image_guidance_scale,
|
| 83 |
num_inference_steps=num_steps,
|
|
|
|
| 78 |
prompt=edit_prompt,
|
| 79 |
negative_prompt=DEFAULT_NEGATIVE_PROMPT,
|
| 80 |
image=input_image,
|
| 81 |
+
height=generate_size,
|
| 82 |
+
width=generate_size,
|
| 83 |
guidance_scale=guidance_scale,
|
| 84 |
image_guidance_scale=image_guidance_scale,
|
| 85 |
num_inference_steps=num_steps,
|