Update app.py

app.py

@@ -52,7 +52,7 @@ def calculate_optimal_dimensions(image: Image.Image):
 
     return width, height
 
-@spaces.GPU
+@spaces.GPU(duration=150)
 def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
     image = edit_images["background"]
     width, height = calculate_optimal_dimensions(image)