Update app.py

app.py

@@ -15,31 +15,32 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 #predictor = gr.State(None)
 
-#def init_predictor(task_type: str):
-from skyreelsinfer import TaskType
-from skyreelsinfer.offload import OffloadConfig
-from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
-from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError
+def init_predictor(task_type: str):
+    from skyreelsinfer import TaskType
+    from skyreelsinfer.offload import OffloadConfig
+    from skyreelsinfer.skyreels_video_infer import SkyReelsVideoInfer
+    from huggingface_hub.utils import RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError
 #try:
-predictor = SkyReelsVideoInfer(
-        task_type=TaskType.I2V, # if task_type == "i2v" else TaskType.T2V,
-        model_id="Skywork/skyreels-v1-Hunyuan-i2v",
-        quant_model=True,
-        is_offload=True,
-        offload_config=OffloadConfig(
-            high_cpu_memory=True,
-            parameters_level=True,
-        ),
-        use_multiprocessing=False,
-)
+    predictor = SkyReelsVideoInfer(
+            task_type=TaskType.I2V, # if task_type == "i2v" else TaskType.T2V,
+            model_id="Skywork/skyreels-v1-Hunyuan-i2v",
+            quant_model=True,
+            is_offload=True,
+            offload_config=OffloadConfig(
+                high_cpu_memory=True,
+                parameters_level=True,
+            ),
+            use_multiprocessing=False,
+    )
     #return "Model loaded successfully!", predictor  # Return predictor
+    return predictor
 
 #except (RepositoryNotFoundError, RevisionNotFoundError, EntryNotFoundError) as e:
 #    return f"Error: Model not found. Details: {e}", None
 #except Exception as e:
 #    return f"Error loading model: {e}", None
         
-#init_predictor('i2v')
+predictor = init_predictor('i2v')
 
 @spaces.GPU(duration=80)
 def generate_video(prompt, image):