Update app.py

app.py

@@ -33,8 +33,8 @@ pipe.load_lora_weights(custom_model, use_auth_token=True)
 pipe.enable_model_cpu_offload()
 
 def infer(image_in, prompt):
-    prompt = "aerial view, a futuristic research complex in a bright foggy jungle, hard lighting"
-    negative_prompt = 'low quality, bad quality, sketches'
+    prompt = prompt
+    negative_prompt = ""
 
     image = load_image("https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/hf-logo.png")
 
@@ -50,7 +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]
 
 with gr.Blocks() as demo:
     with gr.Column():