add download button (#2)

app.py

@@ -193,7 +193,7 @@ def inference(image, background_enhance, face_upsample, upscale, codeformer_fide
     imwrite(restored_img, str(save_path))
 
     restored_img = cv2.cvtColor(restored_img, cv2.COLOR_BGR2RGB)
-    return restored_img
+    return restored_img, save_path
 
 
 
@@ -242,6 +242,7 @@ demo = gr.Interface(
         gr.Slider(0, 1, value=0.5, step=0.01, label='Codeformer_Fidelity: 0 for better quality, 1 for better identity')
     ], [
         gr.outputs.Image(type="numpy", label="Output"),
+        gr.outputs.File(label="Download the output")
     ],
     title=title,
     description=description,