Update app.py

app.py

@@ -35,7 +35,7 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     torch_dtype=torch.bfloat16
 )
 
-pipe.to("cpu")
+# pipe.to("cpu")
 
 pipe.vae.enable_tiling()
 pipe.vae.enable_slicing()
@@ -50,6 +50,13 @@ if file_name is not None:
     image = Image.open(file_name)
     col1.image(image, use_container_width=True)
 
+    seed = 42
+
+    if torch.cuda.is_available():
+        generator = torch.Generator(device="cuda").manual_seed(seed)
+    else:
+        generator = torch.Generator().manual_seed(seed)
+
     video = pipe(
         prompt=prompt,
         image=image,
@@ -57,7 +64,7 @@ if file_name is not None:
         num_inference_steps=50,
         num_frames=49,
         guidance_scale=6,
-        generator=torch.Generator(device="cuda").manual_seed(42),
+        generator=generator,
     ).frames[0]
 
     col2.video(video, use_container_width=True)