msg

app.py

@@ -42,21 +42,23 @@ def infer(image):
    "RGB"
    )
 
+   out_img.save('output.png')
+
    out = {}
-   out.update( {'input image size': (nx,ny) } )
-   out.update( {'output image size': (nxo,nyo) } )
+   out.update( {'input image size x': nx } )
+   out.update( {'output image size x': nxo } )
+   out.update( {'input image size y': ny } )
+   out.update( {'output image size y': nyo } )
 
-   return (pd.DataFrame(data=out.values(), index=out.keys()).transpose(), img,out_img)
+   return (pd.DataFrame(data=out.values(), index=out.keys()).transpose(), img,out_img, 'output.png')
 	
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1609.05158' target='_blank'>Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network</a></p><center> <a href='https://keras.io/examples/vision/super_resolution_sub_pixel/' target='_blank'>Image Super-Resolution using an Efficient Sub-Pixel CNN</a></p>"
 
-# examples = [['examples/2000-04-28-18-21-24_L5_rgb.jpg'],['examples/2000-08-02-18-23-18_L5_rgb.jpg'],
-#             ['examples/2000-08-18-18-23-46_L5_rgb.jpg'],['examples/2000-09-19-18-24-18_L5_rgb.jpg'],['examples/2000-10-21-18-24-43_L5_rgb.jpg']]
-
 examples= [[l] for l in glob('examples/tiles/*.jpg')]
-out1 = gr.outputs.Dataframe(label='Summary', headers=["Input (px)", "Output (px)"], type='pandas')
+out1 = gr.outputs.Dataframe(label='Summary', headers=["Input X (px)", "Output X (px)", "Input Y (px)", "Output Y (px)"], type='pandas')
 out2 = gr.outputs.Image(label="Cropped input image", type='pil')
 out3 = gr.outputs.Image(label="Super-resolution x3 image", type='pil')
+out4 = gr.outputs.File(label='Click to download super-resolved image')
 
 iface = gr.Interface(
 	fn=infer,
@@ -64,6 +66,6 @@ iface = gr.Interface(
 	description = "This space is a demo of Satellite image Super-Resolution using a Sub-Pixel Convolutional Neural Network",
 	article = article,
 	inputs=gr.inputs.Image(label="Input Image"),
-	outputs=[out1,out2,out3],
+	outputs=[out1,out2,out3,out4],
 	examples=examples,
 ).launch()