Update app.py

app.py

@@ -11,13 +11,13 @@ from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
 
-predictor = None
-task_type = None
+#predictor = None
+#task_type = None
 
-@spaces.GPU(duration=120)
-def init_predictor():
-    global predictor
-    predictor = SkyReelsVideoSingleGpuInfer(
+#@spaces.GPU(duration=120)
+#def init_predictor():
+#    global predictor
+predictor = SkyReelsVideoSingleGpuInfer(
         task_type= TaskType.I2V,
         model_id="Skywork/SkyReels-V1-Hunyuan-I2V",
         quant_model=True,
@@ -27,7 +27,7 @@ def init_predictor():
             parameters_level=True,
             compiler_transformer=False,
         )
-    )
+)
     
 @spaces.GPU(duration=80)
 def generate_video(prompt, seed, image=None):
@@ -49,7 +49,7 @@ def generate_video(prompt, seed, image=None):
     }
     assert image is not None, "please input image"
     kwargs["image"] = load_image(image=image)
-    global predictor
+    #global predictor
     output = predictor.inference(kwargs)
     save_dir = f"./result/{task_type}"
     os.makedirs(save_dir, exist_ok=True)
@@ -77,6 +77,6 @@ def create_gradio_interface():
 if __name__ == "__main__":
     #import multiprocessing
     #multiprocessing.freeze_support()
-    init_predictor()
+    #init_predictor()
     demo = create_gradio_interface()
     demo.launch()