Update app.py

app.py

@@ -132,13 +132,15 @@ def tiled_flux_img2img(pipe, prompt, image, strength, steps, guidance, generator
                 if x > 0 or y > 0:
                     # Simple linear blend on overlaps
                     mask = Image.new('L', (tile_w, tile_h), 255)
+                    effective_overlap_x = min(overlap, tile_w)
+                    effective_overlap_y = min(overlap, tile_h)
                     if x > 0:
-                        for i in range(overlap):
+                        for i in range(effective_overlap_x):
                             for j in range(tile_h):
                                 mask.putpixel((i, j), int(255 * (i / overlap)))
                     if y > 0:
                         for i in range(tile_w):
-                            for j in range(overlap):
+                            for j in range(effective_overlap_y):
                                 mask.putpixel((i, j), int(255 * (j / overlap)))
                     output.paste(gen_tile, paste_box, mask)
                 else:
@@ -488,5 +490,4 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - FLUX") as demo:
     """)
 
 if __name__ == "__main__":
-    demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860)
-</xai:function_call
+    demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860)