Update app.py

app.py

@@ -244,9 +244,10 @@ def infer(
     if not all_inputs:
         raise gr.Error("Please upload at least one image or a background image.")
 
+    prompt = prompt +". Integrate the product from Image 1 onto Image 2 as the background, ensuring seamless blending with appropriate lighting and shadows" if len(all_inputs) > 1 else prompt
     result = pipe(
         image=all_inputs,
-        prompt=prompt+". Integrate the product from Image 1 onto Image 2 as the background, ensuring seamless blending with appropriate lighting and shadows",
+        prompt=prompt,
         width=width,
         height=height,
         num_inference_steps=num_inference_steps,