Update app.py

app.py

@@ -36,7 +36,7 @@ def infer(image_in, prompt):
     prompt = prompt
     negative_prompt = ""
 
-    image = load_image("https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/hf-logo.png")
+    image = load_image(image_in)
 
     controlnet_conditioning_scale = 0.5  # recommended for good generalization
 
@@ -50,9 +50,9 @@ def infer(image_in, prompt):
         prompt, negative_prompt=negative_prompt, image=image, controlnet_conditioning_scale=controlnet_conditioning_scale,
         ).images
 
-    #images[0].save(f"hug_lab.png")
+    images[0].save(f"hug_lab.png")
 
-    return images[0]
+    return f"hug_lab.png"
 
 with gr.Blocks() as demo:
     with gr.Column():