52Hz commited on
Commit
3676944
1 Parent(s): b532ce8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
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
- [gr.inputs.Image(type="pil", label="Input")],
31
- gr.outputs.Image(type="filepath", label="Output"),
32
  title=title,
33
  description=description,
34
  article=article,
35
  allow_flagging=False,
36
  allow_screenshot=False,
37
  examples=examples
38
- ).launch(debug=True)
 
 
 
 
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)