Update app.py

app.py

@@ -174,7 +174,7 @@ def generate_input_image(image_path,input_choice):
     return conditioning_image
 
 @spaces.GPU
-def generate_output_image(prompt,input_choice):
+def generate_output_image(prompt,conditioning_image,input_choice):
     #---------------------
     if input_choice == "Edge image":
         pipe = pipe_edge
@@ -270,7 +270,7 @@ with gr.Blocks() as demo:
     )
     generate_btn.click(
         fn=generate_output_image,
-        inputs=[prompt_input,image_dropdown],
+        inputs=[prompt_input,input_preview,image_dropdown],
         outputs=[output_image]
     )