Update app.py

app.py

@@ -56,7 +56,7 @@ random_seed = 42
 video_length = 201
 W = 1024
 H = W
-device = "cpu"
+device = "cuda"
 
 def get_pipe_image_and_video_predictor():
     vae = AutoencoderKLWan.from_pretrained("./model/vae", torch_dtype=torch.float16)
@@ -206,7 +206,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
                 height=height,
                 width=width,
                 num_inference_steps=int(num_inference_steps),
-                generator=torch.Generator(device="cuda").manual_seed(random_seed),
+                generator=torch.Generator(device=device).manual_seed(random_seed),
                 iterations=int(dilation_iterations)
         ).frames[0]
 
@@ -218,7 +218,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
     clip.write_videofile(video_file, codec='libx264', audio=False, verbose=False, logger=None)
     return video_file
 
-
+@spaces.GPU(duration=200)
 def track_video(n_frames, video_state):
 
     input_points = video_state["input_points"]