Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
-import os
+import sys
 import time
+import os
 import random
 
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
+# Create the gr.State component *outside* the gr.Blocks context
+predictor_state = gr.State(None)
+
 def get_transformer_model_id(task_type: str) -> str:
     if task_type == "i2v":
         return "Skywork/skyreels-v1-Hunyuan-i2v"
@@ -113,12 +117,12 @@ with gr.Blocks() as demo:
     load_model_button.click(
         fn=init_predictor,
         inputs=[task_type_dropdown],
-        outputs=[model_status, "state"],  # Output to a hidden state
+        outputs=[model_status, predictor_state],  # Output to the gr.State
     )
 
     submit_button.click(
         fn=generate_video,
-        inputs=[prompt, seed, image, task_type_dropdown, "state"],  # Input from state
+        inputs=[prompt, seed, image, task_type_dropdown, predictor_state],  # Input from gr.State
         outputs=[output_video, output_params],
     )