Update app.py

app.py

@@ -57,7 +57,7 @@ def generate_images(canvas,output_type):
     input_img = get_canny_filter(input_img) if output_type == "Canny" else input_img
     input_img = ImageOps.invert(input_img)
     input_img.save("tmp.png")
-    return "tmp.png"
+    return "tmp.png","tmp.png"
 def placeholder_fn(axis):
     pass
 with gr.Blocks() as b:
@@ -68,6 +68,7 @@ with gr.Blocks() as b:
             pose_template = gr.Radio(choices=["regular", "ballet", "handstand", "split", "kick", "chilling"], value="regular", label="Pose template")
             output_type = gr.Radio(choices=["Depth", "Canny"], value="Depth", label="Image Output")
             run_button = gr.Button("Generate")
+            out_file=gr.File()
             out_im = gr.Image(type='filepath',show_share_button=True)
     
     rotation_axis.change(fn=placeholder_fn,
@@ -82,7 +83,7 @@ with gr.Blocks() as b:
                             _js=js_pose_template)
     run_button.click(fn=generate_images,
                      inputs=[canvas,output_type],
-                     outputs=[out_im],
+                     outputs=[out_im,out_file],
                      _js=get_js_image)
     b.load(None,None,None,_js=load_js)