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",
|