Update app.py

app.py

@@ -45,19 +45,24 @@ def duplicate_horizontally(img):
 # Load the mask image
 mask = Image.open("mask_square.png")
 
+def crop_input(image):
+    cropped_image = square_center_crop(image)
+    logo_dupli = duplicate_horizontally(cropped_image)
+    return logo_dupli
+
 @spaces.GPU
 def generate(image, prompt_user, progress=gr.Progress(track_tqdm=True)):
     prompt_structure = "The two-panel image showcases the logo of a brand, [LEFT] the left panel is showing the logo [RIGHT] the right panel has this logo applied to "
     prompt = prompt_structure + prompt_user
 
-    cropped_image = square_center_crop(image)
-    logo_dupli = duplicate_horizontally(cropped_image)
+    #cropped_image = square_center_crop(image)
+    #logo_dupli = duplicate_horizontally(cropped_image)
 
     out = pipe(
         prompt=prompt,
-        image=logo_dupli,
+        image=image,
         mask_image=mask,
-        guidance_scale=6,
+        guidance_scale=3.75,
         height=768,
         width=1536,
         num_inference_steps=28,
@@ -81,6 +86,9 @@ with gr.Blocks() as demo:
                 type="pil",
                 height=384
             )
+            cropped_image = gr.Image(
+                visible=False
+            )
             prompt_input = gr.Textbox(
                 label="Where should the logo be applied?",
                 placeholder="e.g., a coffee cup on a wooden table",
@@ -103,8 +111,12 @@ with gr.Blocks() as demo:
 
     # Set up the click event
     generate_btn.click(
+        fn=crop_input,
+        inputs=[input_image],
+        outputs=[cropped_image]
+    ).then(
         fn=generate,
-        inputs=[input_image, prompt_input],
+        inputs=[cropped_image, prompt_input],
         outputs=[output_image, output_side]
     )