Change download to gr.File type

modules/app.py

@@ -13,17 +13,10 @@ render_factor=35
 
 
 def colorize_image(image):
-    # now = datetime.now().strftime("%Y%m%d-%H%M%S-%f")
     if not os.path.exists("./output"):
         os.makedirs("./output")
-    # if image is not None:
-    #     image.save(f"./output/{now}-input.jpg")    
     model.predict(image.name)
-    return './output/DeOldify/'+Path(image.name).stem+".png"
-
-
-def download_image(image):
-    return image
+    return './output/DeOldify/'+Path(image.name).stem+".png", './output/DeOldify/'+Path(image.name).stem+".png"
 
 
 def create_interface():
@@ -34,8 +27,7 @@ def create_interface():
             colorize_button = gr.Button("Colorize!")
             colorize_output = gr.Image(type="file")
             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=[])
+            colorize_button.click(colorize_image, inputs=colorize_input, outputs=[colorize_output, gr.File(label="Download photo!")],)
     enhancer.launch()
 
 def run_code():