Update app.py

app.py

@@ -10,14 +10,11 @@ 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
 
-#@spaces.GPU(duration=120)
+@spaces.GPU(duration=120)
 #def init_predictor():
 #    global predictor
 predictor = SkyReelsVideoSingleGpuInfer(