Update app.py
Browse files
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
|
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,
|