Update app.py

app.py

@@ -9,6 +9,8 @@ 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")
+# Choose dtype based on device
+dtype = torch.float16 if device.type == "cuda" else torch.float32
 import tempfile
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import export_to_video
@@ -16,9 +18,9 @@ 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 or CPU
+# Load pipeline in appropriate precision on GPU or CPU
 pipe = StableVideoDiffusionPipeline.from_pretrained(
-    MODEL, torch_dtype=torch.float16
+    MODEL, torch_dtype=dtype
 ).to(device)
 
 def infer(first_image, last_image, prompt, guidance=7.5, frames=25):
@@ -50,4 +52,4 @@ demo = gr.Interface(
 )
 
 # Enable the REST API
-demo.queue(concurrency_count=1).launch(show_api=True)
+demo.queue(default_concurrency_limit=1).launch(show_api=True)