Update app.py

app.py

@@ -8,14 +8,15 @@ from PIL import Image
  # os.environ["CUDA_VISIBLE_DEVICES"] = ""
 os.environ["SAFETENSORS_FAST_GPU"] = "1"
 os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
-import torch
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+#import torch
+#device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 # Create the gr.State component *outside* the gr.Blocks context
 
 #predictor = gr.State(None)
 
 def init_predictor(task_type: str):
+    import torch
     from skyreelsinfer import TaskType
     from skyreelsinfer.offload import OffloadConfig
     from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer