Update app.py

app.py

@@ -9,11 +9,10 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
         torch_dtype=torch.bfloat16
     )
 
-@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")
-    pipe.to("cuda")
+    pipe.to("cpu")
     generator = torch.Generator(device=device).manual_seed(42)
     
     video = pipe(