add more time to zerogpu

app.py

@@ -12,11 +12,11 @@ import stone
 import os
 import spaces
 
-api_key = os.getenv("AccessTokenSD3")
+access_token = os.getenv("AccessTokenSD3")
 
 
 from huggingface_hub import login
-login(token = api_key)
+login(token = access_token)
 
 
 # Define model initialization functions
@@ -52,12 +52,9 @@ def load_model(model_name):
             variant="fp16"
         ).to("cuda")
     elif model_name == "stabilityai/stable-diffusion-3-medium-diffusers":
-        if api_key is None:
-            raise ValueError("Hugging Face token is required to access this model")
         pipeline = StableDiffusion3Pipeline.from_pretrained(
             model_name, 
             torch_dtype=torch.float16,
-            use_auth_token=api_key
         ).to("cuda")
     else:
         raise ValueError("Unknown model name")
@@ -123,7 +120,7 @@ def skintoneplot(hex_codes):
             ax.add_patch(plt.Rectangle((0, 0), 1, 1, color=sorted_hex_codes[i]))
     return fig
 
-@spaces.GPU
+@spaces.GPU(duration=400)
 def generate_images_plots(prompt, model_name):
     global pipeline_text2image
     pipeline_text2image = load_model(model_name)