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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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