Update app.py
Browse files
app.py
CHANGED
@@ -172,13 +172,9 @@ def fn_diffusion(
|
|
172 |
demo = gr.Blocks(css=open('app.css').read())
|
173 |
|
174 |
with demo:
|
175 |
-
gr.HTML(open('app_header.html').read())
|
176 |
-
|
177 |
-
if not try_cuda or not torch.cuda.is_available():
|
178 |
-
gr.HTML('<div class="alert alert-warning" role="alert" style="color:red"><b>Warning: GPU not available! Diffusion will be slow.</b></div>')
|
179 |
|
180 |
# Input image control
|
181 |
-
input_image = gr.Image(
|
182 |
# Combined mask controls
|
183 |
bt_masks = gr.Button("Compute Masks")
|
184 |
with gr.Row():
|
|
|
172 |
demo = gr.Blocks(css=open('app.css').read())
|
173 |
|
174 |
with demo:
|
|
|
|
|
|
|
|
|
175 |
|
176 |
# Input image control
|
177 |
+
input_image = gr.Image(type='pil', label="Input Image")
|
178 |
# Combined mask controls
|
179 |
bt_masks = gr.Button("Compute Masks")
|
180 |
with gr.Row():
|