p2p add download_path

app_haircolor_pix2pix.py

@@ -59,10 +59,9 @@ def image_to_image(
     
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
     enhanced_image = enhance_image(generated_image, False)
+    run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
     return enhanced_image, time_cost_str
 
-    return generated_image, time_cost_str
-
 def get_time_cost(run_task_time, time_cost_str):
     now_time = int(time.time()*1000)
     if run_task_time == 0:
@@ -98,6 +97,7 @@ def create_demo() -> gr.Blocks:
                 input_image = gr.Image(label="Input Image", type="pil")
             with gr.Column():
                 restored_image = gr.Image(label="Restored Image", type="pil", interactive=False)
+                download_path = gr.File(label="Download the output image", interactive=False)
             with gr.Column():
                 origin_area_image = gr.Image(label="Origin Area Image", type="pil", interactive=False)
                 generated_image = gr.Image(label="Generated Image", type="pil", interactive=False)
@@ -114,7 +114,7 @@ def create_demo() -> gr.Blocks:
         ).success(
             fn=restore_result,
             inputs=[croper, category, generated_image],
-            outputs=[restored_image],
+            outputs=[restored_image, download_path],
         )
 
     return demo