Update app.py

app.py

@@ -18,7 +18,8 @@ import spaces
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "vdo/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16"
 )
-pipe.to("cuda")
+pipe.to("cuda:0" if torch.cuda.is_available() else "cpu")
+#device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 max_64_bit_int = 2**63 - 1