Update app.py

app.py

@@ -23,11 +23,11 @@ def infer(img):
     os.system('python main_test_swinir.py')
     res=Image.open("./results/data.png")
     #return res
-    return "./results/data.png"
+    return "./results/data.png","./results/data.png"
 
 
 inputs = [gr.inputs.Image(type='pil', label="Original Image")]
-outputs = [gr.outputs.Image(type="file", label="output")]
+outputs = [gr.outputs.Image(type="file", label="output"), gr.outputs.File(label="download")]
 title = "SwinIR: Image Restoration Using Swin Transformer,Super-Resolution part "
 description = "Gradio demo for SwinIR: Super-Resolution part. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
 article = "<p style='text-align: center'><a href='https://arxiv.org/pdf/2108.10257.pdf' target='_blank'>SwinIR: Image Restoration Using Swin Transformer</a> | <a href='https://github.com/JingyunLiang/SwinIR' target='_blank'>Github Repo</a></p>"