Update app.py

app.py

@@ -14,10 +14,11 @@ vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=to
 pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 flow_shift = 5.0  # 5.0 for 720P, 3.0 for 480P
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
-pipe.to("cuda")
+
 
 @spaces.GPU()
 def generate(prompt):
+    pipe.to("cuda")
     output = pipe(
         prompt=prompt,
         # negative_prompt=negative_prompt,