Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,14 +25,17 @@ 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 |
-
gr.Interface(
|
29 |
-
inference,
|
30 |
-
|
31 |
-
gr.
|
32 |
title=title,
|
33 |
description=description,
|
34 |
article=article,
|
35 |
allow_flagging=False,
|
36 |
allow_screenshot=False,
|
37 |
examples=examples
|
38 |
-
)
|
|
|
|
|
|
|
|
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
|
31 |
+
outputs=gr.Image(type="pil", label="Output"), # Updated to gr.Image
|
32 |
title=title,
|
33 |
description=description,
|
34 |
article=article,
|
35 |
allow_flagging=False,
|
36 |
allow_screenshot=False,
|
37 |
examples=examples
|
38 |
+
)
|
39 |
+
|
40 |
+
# Launch the interface with debugging
|
41 |
+
interface.launch(debug=True)
|