Fix download button

modules/app.py

@@ -22,6 +22,10 @@ def colorize_image(image):
     return './output/DeOldify/'+Path(image.name).stem+".png"
 
 
+def download_image(image):
+    return image
+
+
 def create_interface():
     with gr.Blocks() as enhancer:
         gr.Markdown("Colorize old black & white photos")
@@ -29,8 +33,9 @@ def create_interface():
             colorize_input = gr.Image(type="file")
             colorize_button = gr.Button("Colorize!")
             colorize_output = gr.Image(type="file")
-            download_colorize_button = gr.outputs.File(label="Download colorized image!")
+            download_button = gr.Button("Download!")
             colorize_button.click(colorize_image, inputs=colorize_input, outputs=colorize_output)
+            download_button.click(download_image, inputs=colorize_output, outputs=[])
     enhancer.launch()
 
 def run_code():