Update app.py

app.py

@@ -6,8 +6,6 @@ import gradio as gr
 import random
 import tqdm
 
-# HF_TOKEN = os.environ.get("HF_TOKEN") or True
-
 # Enable TQDM progress tracking
 tqdm.monitor_interval = 0
 
@@ -17,12 +15,11 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     torch_dtype=torch.float16, 
     custom_pipeline="lpw_stable_diffusion_xl",
     use_safetensors=True, 
-    # use_auth=HF_TOKEN,
 )
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 # Function to generate an image
-@spaces.GPU(duration=120)  # Adjust the duration as needed
+@spaces.GPU()  # Adjust the duration as needed
 def generate_image(prompt, negative_prompt, use_defaults, width, height, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
     pipe.to('cuda')  # Move the model to GPU when the function is called