Update app.py

app.py

@@ -55,8 +55,9 @@ def generate_images(canvas,output_type):
         'RGB').resize((512, 512))
     input_img = input_img.filter(ImageFilter.GaussianBlur(radius=2))
     input_img = get_canny_filter(input_img) if output_type == "Canny" else input_img
-    input_img = ImageOps.invert(input_img)    
-    return input_img
+    input_img = ImageOps.invert(input_img)
+    input_img.save("tmp.png")
+    return "tmp.png"
 def placeholder_fn(axis):
     pass
 with gr.Blocks() as b:
@@ -67,7 +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_im = gr.Image(type='filepath',source='upload',interactive=True)
+            out_im = gr.Image(type='filepath',show_share_button=True,interactive=True)
     
     rotation_axis.change(fn=placeholder_fn,
                             inputs=[rotation_axis],