Update app.py

app.py

@@ -164,7 +164,7 @@ def center_crop_resize(input_video_path, target_width=720, target_height=480):
     return temp_video_path
 
 
-
+@spaces.GPU(duration=120)
 def infer(
     prompt: str,
     num_inference_steps: int,
@@ -174,6 +174,7 @@ def infer(
 ):
     if seed == -1:
         seed = random.randint(0, 2**8 - 1)
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     pipe.to(device)
     video_pt = pipe(
         prompt=prompt + ", isolated background",
@@ -278,7 +279,7 @@ with gr.Blocks() as demo:
                 download_alpha_video_button = gr.File(label="📥 Download Alpha Video", visible=False)
                 seed_text = gr.Number(label="Seed Used for Video Generation", visible=False)
 
-
+    @spaces.GPU(duration=120)
     def generate(
         prompt,
         seed_value,