update params
Browse files
app.py
CHANGED
|
@@ -57,6 +57,8 @@ def inpaint(
|
|
| 57 |
padding_left=0,
|
| 58 |
padding_right=0,
|
| 59 |
prompt="",
|
|
|
|
|
|
|
| 60 |
):
|
| 61 |
background, mask = prepare_image_and_mask(
|
| 62 |
image, padding_top, padding_bottom, padding_left, padding_right
|
|
@@ -68,8 +70,8 @@ def inpaint(
|
|
| 68 |
width=background.width,
|
| 69 |
image=background,
|
| 70 |
mask_image=mask,
|
| 71 |
-
num_inference_steps=
|
| 72 |
-
guidance_scale=
|
| 73 |
).images[0]
|
| 74 |
|
| 75 |
result = result.convert("RGBA")
|
|
@@ -120,6 +122,8 @@ outpaint_tab = gr.Interface(
|
|
| 120 |
gr.Number(label="padding left"),
|
| 121 |
gr.Number(label="padding right"),
|
| 122 |
gr.Text(label="prompt"),
|
|
|
|
|
|
|
| 123 |
],
|
| 124 |
outputs=["image"],
|
| 125 |
api_name="outpainting",
|
|
|
|
| 57 |
padding_left=0,
|
| 58 |
padding_right=0,
|
| 59 |
prompt="",
|
| 60 |
+
num_inference_steps=28,
|
| 61 |
+
guidance_scale=50,
|
| 62 |
):
|
| 63 |
background, mask = prepare_image_and_mask(
|
| 64 |
image, padding_top, padding_bottom, padding_left, padding_right
|
|
|
|
| 70 |
width=background.width,
|
| 71 |
image=background,
|
| 72 |
mask_image=mask,
|
| 73 |
+
num_inference_steps=num_inference_steps,
|
| 74 |
+
guidance_scale=guidance_scale,
|
| 75 |
).images[0]
|
| 76 |
|
| 77 |
result = result.convert("RGBA")
|
|
|
|
| 122 |
gr.Number(label="padding left"),
|
| 123 |
gr.Number(label="padding right"),
|
| 124 |
gr.Text(label="prompt"),
|
| 125 |
+
gr.Number(value=50, label="num_inference_steps"),
|
| 126 |
+
gr.Number(value=28, label="guidance_scale"),
|
| 127 |
],
|
| 128 |
outputs=["image"],
|
| 129 |
api_name="outpainting",
|