add duration

app.py

@@ -297,9 +297,9 @@ def _encode_prompt(prompt: str):
 # -----------------------------
 # Inference functions
 # -----------------------------
-@spaces.GPU
+@spaces.GPU(duration=300)
 @torch.no_grad()
-def generate_image_stack_edits(text_prompt, n_edits, input_image):
+def generate_image_stack_edits(text_prompt, n_edits, input_image, progress=gr.Progress(track_tqdm=True)):
     """
     Compute n_edits images on a single GPU for slider values in (0,1],
     return (list_of_images, first_image) so the UI shows immediately.
@@ -350,8 +350,8 @@ def generate_image_stack_edits(text_prompt, n_edits, input_image):
     first = results[0] if results else None
     return results, first
 
-@spaces.GPU
-def generate_single_image(text_prompt, slider_value, input_image):
+@spaces.GPU(duration=80)
+def generate_single_image(text_prompt, slider_value, input_image, progress=gr.Progress(track_tqdm=True)):
     if not input_image or not text_prompt or text_prompt.startswith("Please select"):
         return None