Update app.py

app.py

@@ -11,14 +11,12 @@ from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
 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
 
 def init_predictor():
-    import spaces
-    import torch
-    device="cuda:0" if torch.cuda.is_available() else "cpu" # Pass device to the constructor
     global predictor
     predictor = SkyReelsVideoInfer(
         task_type= TaskType.I2V,
@@ -33,7 +31,7 @@ def init_predictor():
         )
     )
     
-#init_predictor()
+init_predictor()
 
 @spaces.GPU(duration=80)
 def generate_video(prompt, seed, image=None):
@@ -81,6 +79,6 @@ def create_gradio_interface():
         return demo
 
 if __name__ == "__main__":
-    init_predictor()
+    #init_predictor()
     demo = create_gradio_interface()
     demo.launch()