Update app.py

app.py

@@ -46,7 +46,7 @@ def load_lora_model(pipe, lora_model_path):
 pipe = load_lora_model(pipe, lora_model_path)
 
 # Use the @spaces.gpu decorator to ensure compatibility with GPU or CPU as needed
-@spaces.gpu
+@spaces.GPU(duration=65)  # This ensures GPU is allocated for 65 seconds
 def generate(prompt, seed=None):
     generator = torch.manual_seed(seed) if seed is not None else None
     # Generate the image using the prompt