Spaces:
Runtime error
Runtime error
IanNathaniel
commited on
Commit
•
d6abebc
1
Parent(s):
a0d1f70
Update app.py
Browse files
app.py
CHANGED
@@ -46,8 +46,8 @@ def lowlight(image):
|
|
46 |
if not os.path.exists(image_path.replace('/'+image_path.split("/")[-1],'')):
|
47 |
os.makedirs(image_path.replace('/'+image_path.split("/")[-1],''))
|
48 |
|
49 |
-
torchvision.utils.
|
50 |
-
return
|
51 |
|
52 |
|
53 |
title = "Compound Multi-branch Feature Fusion for Image Restoration (Deblur)"
|
@@ -58,7 +58,7 @@ examples = [['data/test_data/01.jpg'], ['data/test_data/02.jpg'], ['data/test_da
|
|
58 |
gr.Interface(
|
59 |
lowlight,
|
60 |
[gr.inputs.Image(type="file", label="Input")],
|
61 |
-
gr.outputs.Image(type="
|
62 |
title=title,
|
63 |
description=description,
|
64 |
article=article,
|
|
|
46 |
if not os.path.exists(image_path.replace('/'+image_path.split("/")[-1],'')):
|
47 |
os.makedirs(image_path.replace('/'+image_path.split("/")[-1],''))
|
48 |
|
49 |
+
grid = torchvision.utils.make_grid(enhanced_image)
|
50 |
+
return grid
|
51 |
|
52 |
|
53 |
title = "Compound Multi-branch Feature Fusion for Image Restoration (Deblur)"
|
|
|
58 |
gr.Interface(
|
59 |
lowlight,
|
60 |
[gr.inputs.Image(type="file", label="Input")],
|
61 |
+
gr.outputs.Image(type="pil", label="Output"),
|
62 |
title=title,
|
63 |
description=description,
|
64 |
article=article,
|