Update app.py

app.py

@@ -1,4 +1,4 @@
-#This code is from: https://huggingface.co/spaces/diffusers/controlnet-3d-pose
+#This code is largely based on: https://huggingface.co/spaces/diffusers/controlnet-3d-pose
 import gradio as gr
 from PIL import Image, ImageFilter, ImageOps
 from io import BytesIO
@@ -57,7 +57,9 @@ 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","tmp.png"
+    out = os.path.abspath("tmp.png")
+    out_url = f'https://omnibus-model-mover.hf.space/file={out}'       
+    return "tmp.png","tmp.png",out
 def placeholder_fn(axis):
     pass
 with gr.Blocks() as b:
@@ -68,6 +70,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_text = gr.Textbox(label="Image URL")
             out_file=gr.File()
             out_im = gr.Image(type='filepath',show_share_button=True)
     
@@ -83,7 +86,7 @@ with gr.Blocks() as b:
                             _js=js_pose_template)
     run_button.click(fn=generate_images,
                      inputs=[canvas,output_type],
-                     outputs=[out_im,out_file],
+                     outputs=[out_im,out_file,out_text],
                      _js=get_js_image)
     b.load(None,None,None,_js=load_js)