Update app.py

app.py

@@ -36,15 +36,6 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
     
     Returns:
         The modified image and seed used for generation.
-    
-    Example:
-        >>> edited_image, used_seed, button_update = infer(
-        ...     input_image=my_image,
-        ...     prompt="Add sunglasses",
-        ...     seed=123,
-        ...     randomize_seed=False,
-        ...     guidance_scale=2.5
-        ... )
     """
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -69,7 +60,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=20)
 def infer_example(input_image, prompt):
     image, seed, _ = infer(input_image, prompt)
     return image, seed