Update app.py

app.py

@@ -7,6 +7,8 @@ if not hasattr(hf_hub, "model_info"):
 
 import gradio as gr
 import torch
+# Determine device: use GPU if available, otherwise CPU
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 import tempfile
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import export_to_video
@@ -14,10 +16,10 @@ from diffusers.utils import export_to_video
 # Use the official SVD-XT img2vid-xt model
 MODEL = "stabilityai/stable-video-diffusion-img2vid-xt"
 
-# Load pipeline in half-precision on GPU
+# Load pipeline in half-precision on GPU or CPU
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     MODEL, torch_dtype=torch.float16
-).to("cuda")
+).to(device)
 
 def infer(first_image, last_image, prompt, guidance=7.5, frames=25):
     # Generate the in-between frames