Update app.py

app.py

@@ -11,14 +11,10 @@ from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
 
-@spaces.GPU(duration=5)
-def init_gpu():
-    import torch
-    device="cuda:0" if torch.cuda.is_available() else "cpu" # Pass device to the constructor
-
 predictor = None
 task_type = None
 
+@spaces.GPU()
 def init_predictor():
     global predictor
     predictor = SkyReelsVideoInfer(
@@ -82,7 +78,6 @@ def create_gradio_interface():
 if __name__ == "__main__":
     #import multiprocessing
     #multiprocessing.freeze_support()
-    init_gpu()
     init_predictor()
     demo = create_gradio_interface()
     demo.launch()