Update app.py

app.py

@@ -8,12 +8,14 @@ from PIL import Image
 from diffusers import FluxKontextPipeline
 from diffusers.utils import load_image
 
+# down to 22 steps to try and keep this ~<30 seconds so it will generally work in claude.ai - which doesn't reset timeout with notifications.
+
 MAX_SEED = np.iinfo(np.int32).max
 
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 
 @spaces.GPU
-def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=22, progress=gr.Progress(track_tqdm=True)):
     """
     Perform image editing using the FLUX.1 Kontext pipeline.
     
@@ -60,7 +62,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
         ).images[0]
     return image, seed, gr.Button(visible=True)
 
-@spaces.GPU(duration=20)
+@spaces.GPU(duration=25)
 def infer_example(input_image, prompt):
     image, seed, _ = infer(input_image, prompt)
     return image, seed