Update app.py

app.py

@@ -9,7 +9,7 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
         torch_dtype=torch.bfloat16
     )
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=250)
 def generate_video(prompt, image):
     # Ensure the generator is on the same device as the model
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -20,7 +20,7 @@ def generate_video(prompt, image):
         prompt=prompt,
         image=image,
         num_videos_per_prompt=1,
-        num_inference_steps=50,
+        num_inference_steps=15,
         num_frames=49,
         guidance_scale=6,
         generator=generator,