-,

app.py

@@ -15,13 +15,14 @@ import uuid
 import random
 from huggingface_hub import hf_hub_download
 
-#gradio.helpers.CACHED_FOLDER = '/data/cache'
+# Check if GPU is available
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load the pipeline with authentication token
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid-xt",
     torch_dtype=torch.float16,
-    variant="fp16",
+    variant="fp16"
 )
 pipe.to("cuda")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)