Update app.py
Browse files
app.py
CHANGED
|
@@ -112,7 +112,7 @@ def infer(
|
|
| 112 |
if d_bckg:
|
| 113 |
image=remove(image)
|
| 114 |
|
| 115 |
-
|
| 116 |
return image, seed
|
| 117 |
|
| 118 |
|
|
@@ -160,8 +160,12 @@ with gr.Blocks(css=css) as demo:
|
|
| 160 |
step=0.05,
|
| 161 |
value=1,
|
| 162 |
)
|
| 163 |
-
|
| 164 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
with gr.Row():
|
| 166 |
prompt = gr.Text(
|
| 167 |
label="Prompt",
|
|
|
|
| 112 |
if d_bckg:
|
| 113 |
image=remove(image)
|
| 114 |
|
| 115 |
+
#pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config, rescale_betas_zero_snr=True)
|
| 116 |
return image, seed
|
| 117 |
|
| 118 |
|
|
|
|
| 160 |
step=0.05,
|
| 161 |
value=1,
|
| 162 |
)
|
| 163 |
+
with gr.Row():
|
| 164 |
+
d_bckg=gr.Checkbox(label="Delete Background", value=False)
|
| 165 |
+
ddim_use=gr.Checkbox(label="Enable DDIMScheduler", value=False)
|
| 166 |
+
distill_vae=gr.Checkbox(label="Use tiny VAE with distill model", value=True)
|
| 167 |
+
|
| 168 |
+
# pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config, rescale_betas_zero_snr=True)
|
| 169 |
with gr.Row():
|
| 170 |
prompt = gr.Text(
|
| 171 |
label="Prompt",
|