Update app.py

app.py

@@ -8,11 +8,7 @@ import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
 from diffusers.utils import export_to_video
 
-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_xl.scheduler.config)
-pipe_xl.enable_model_cpu_offload()
-pipe_xl.to("cuda")
+
 
 def convert_mp4_to_frames(video_path, duration=3):
     # Read the video file
@@ -58,6 +54,12 @@ def infer(prompt, video_in, denoise_strength):
 
     video = convert_mp4_to_frames(video_in, duration=3)
     video_resized = [Image.fromarray(frame).resize((1024, 576)) for frame in video]
+
+    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_xl.scheduler.config)
+    pipe_xl.enable_model_cpu_offload()
+    pipe_xl.to("cuda")
     video_frames = pipe_xl(prompt, negative_prompt=negative_prompt, video=video_resized, strength=denoise_strength).frames
     del pipe_xl
     torch.cuda.empty_cache()