Update app.py

app.py

@@ -16,16 +16,14 @@ pipe_15 = DiffusionPipeline.from_pretrained(model_15, vae=vae, scheduler=schedul
 #pipe.enable_model_cpu_offload()
 pipe.enable_vae_tiling()
 
-# Apply hidiffusion with a single line of code.
-apply_hidiffusion(pipe)
-apply_hidiffusion(pipe_15)
-
 @spaces.GPU
 def run_hidiffusion(prompt, negative_prompt="", progress=gr.Progress(track_tqdm=True)):
+    apply_hidiffusion(pipe)
     return pipe(prompt, guidance_scale=7.5, height=2048, width=2048, eta=1.0, negative_prompt=negative_prompt, num_inference_steps=25).images[0]
 
 @spaces.GPU
 def run_hidiffusion_15(prompt, negative_prompt="", progress=gr.Progress(track_tqdm=True)):
+    apply_hidiffusion(pipe_15)
     return pipe_15(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt, num_inference_steps=25).images[0]
 
 with gr.Blocks() as demo: