Update app.py

app.py

@@ -109,7 +109,6 @@ class ModelWrapper:
         
     
     @torch.no_grad()
-    @spaces.GPU()
     def inference(self, prompt, seed, height, width, num_images, fast_vae_decode):
         print("Running model inference...")
 
@@ -148,6 +147,7 @@ class ModelWrapper:
 
         return output_image_list, f"Run successfully in {(end_time-start_time):.2f} seconds"
 
+
 def get_x0_from_noise(sample, model_output, alphas_cumprod, timestep):
     alpha_prod_t = alphas_cumprod[timestep].reshape(-1, 1, 1, 1)
     beta_prod_t = 1 - alpha_prod_t
@@ -184,7 +184,7 @@ class SDXLTextEncoder(torch.nn.Module):
         
         return prompt_embeds, pooled_prompt_embeds
 
-
+@spaces.GPU(duration=100)
 def create_demo():
     TITLE = "# DMD2-SDXL Demo"
     model_id = "stabilityai/stable-diffusion-xl-base-1.0"