Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,6 +25,7 @@ description = "Gradio demo for SUNet. SUNet has competitive performance results
|
|
25 |
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2202.14009' target='_blank'>SUNet: Swin Transformer with UNet for Image Denoising</a> | <a href='https://github.com/FanChiMao/SUNet' target='_blank'>Github Repo</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=52Hz_SUNet_AWGN_denoising' alt='visitor badge'></center>"
|
26 |
|
27 |
examples = [['set5/baby.png'], ['set5/bird.png'],['set5/butterfly.png'],['set5/head.png'],['set5/woman.png']]
|
|
|
28 |
interface = gr.Interface(
|
29 |
fn=inference,
|
30 |
inputs=gr.Image(type="pil", label="Input"), # Updated to gr.Image
|
@@ -33,7 +34,6 @@ interface = gr.Interface(
|
|
33 |
description=description,
|
34 |
article=article,
|
35 |
allow_flagging=False,
|
36 |
-
allow_screenshot=False,
|
37 |
examples=examples
|
38 |
)
|
39 |
|
|
|
25 |
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2202.14009' target='_blank'>SUNet: Swin Transformer with UNet for Image Denoising</a> | <a href='https://github.com/FanChiMao/SUNet' target='_blank'>Github Repo</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=52Hz_SUNet_AWGN_denoising' alt='visitor badge'></center>"
|
26 |
|
27 |
examples = [['set5/baby.png'], ['set5/bird.png'],['set5/butterfly.png'],['set5/head.png'],['set5/woman.png']]
|
28 |
+
# Create a Gradio Interface using the updated API
|
29 |
interface = gr.Interface(
|
30 |
fn=inference,
|
31 |
inputs=gr.Image(type="pil", label="Input"), # Updated to gr.Image
|
|
|
34 |
description=description,
|
35 |
article=article,
|
36 |
allow_flagging=False,
|
|
|
37 |
examples=examples
|
38 |
)
|
39 |
|