Update app.py

app.py

@@ -20,7 +20,7 @@ MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 optimize_pipeline_(pipe, image=Image.new("RGB", (512, 512)), prompt='prompt')
 
-@spaces.GPU
+@spaces.GPU(duration=24)
 def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
     """
     Perform image editing using the FLUX.1 Kontext pipeline.
@@ -84,7 +84,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
+@spaces.GPU(duration=25)
 def infer_example(input_image, prompt):
     image, seed, _ = infer(input_image, prompt)
     return image, seed
@@ -171,4 +171,4 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
         outputs = [input_image]
     )
 
-demo.launch(mcp_server=True)
+demo.queue(api_open=True).launch(show_error=True, show_api=True, mcp_server=True)