Update app.py

app.py

@@ -19,14 +19,14 @@ pipe_xl.enable_model_cpu_offload()
 
 def infer(prompt):
     #prompt = "Darth Vader is surfing on waves"
-    pipe.to("cuda")
+    #pipe.to("cuda")
     video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames
     video_path = export_to_video(video_frames)
     print(video_path)
 
     video = [Image.fromarray(frame).resize((1024, 576)) for frame in video_frames]
     del pipe
-    pipe_xl.to("cuda")
+    #pipe_xl.to("cuda")
     video_frames = pipe_xl(prompt, video=video, strength=0.6).frames
     video_path = export_to_video(video_frames, output_video_path="xl_result.mp4")