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,
|