Update app.py

app.py

@@ -10,6 +10,9 @@ from skyreelsinfer import TaskType
 from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
+import torch
+
+device="cuda:0" if torch.cuda.is_available() else "cpu" # Pass device to the constructor
 
 #predictor = None
 #task_type = None