Update app.py

app.py

@@ -9,12 +9,12 @@ from diffusers.utils import export_to_video
 pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dtype=torch.float16)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.enable_model_cpu_offload()
-
+pipe.to("cuda")
 pipe_xl = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_XL", torch_dtype=torch.float16, revision="refs/pr/17")
 pipe_xl.vae.enable_slicing()
 pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe_xl.enable_model_cpu_offload()
-
+pipe_xl.to("cpu")
 
 
 def infer(prompt):