Update app.py

app.py

@@ -18,9 +18,9 @@ from huggingface_hub import hf_hub_download
 #gradio.helpers.CACHED_FOLDER = '/data/cache'
 
 pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "multimodalart/stable-video-diffusion", torch_dtype=torch.float16, variant="fp16"
+    "stabilityai/stable-video-diffusion-img2vid", torch_dtype=torch.float16, variant="fp16"
 )
-pipe.to("cuda")
+pipe.to("cpu")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)