to-be commited on
Commit
0d48277
1 Parent(s): a1caa30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -26,7 +26,7 @@ Gradio demo for <b>Restormer: Efficient Transformer for High-Resolution Image Re
26
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.09881'>Restormer: Efficient Transformer for High-Resolution Image Restoration </a> | <a href='https://github.com/swz30/Restormer'>Github Repo</a></p>"
27
 
28
 
29
- def inference(img, task):
30
  if not os.path.exists('temp'):
31
  os.system('mkdir temp')
32
 
@@ -82,8 +82,6 @@ gr.Interface(
82
  inference,
83
  [
84
  gr.Image(type="pil", label="Input"),
85
- gr.Radio(["Deraining"], default="Denoising", label='task'),
86
-
87
  ],
88
  gr.Image(type="pil", label="cleaned and restored"),
89
  title=title,
 
26
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.09881'>Restormer: Efficient Transformer for High-Resolution Image Restoration </a> | <a href='https://github.com/swz30/Restormer'>Github Repo</a></p>"
27
 
28
 
29
+ def inference(img):
30
  if not os.path.exists('temp'):
31
  os.system('mkdir temp')
32
 
 
82
  inference,
83
  [
84
  gr.Image(type="pil", label="Input"),
 
 
85
  ],
86
  gr.Image(type="pil", label="cleaned and restored"),
87
  title=title,