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():
|