IanNathaniel commited on
Commit
d6abebc
1 Parent(s): a0d1f70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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.save_image(enhanced_image, result_path)
50
- return result_path
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="file", label="Output"),
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,