Update stf_utils.py

stf_utils.py

@@ -78,25 +78,42 @@ class STFPipeline:
                  
     ):
         
-        config_path = os.path.join(stf_path, config_path)
-        checkpoint_path = os.path.join(stf_path, checkpoint_path)
-        work_root_path = os.path.join(stf_path, root_path)
+        self.config_path = os.path.join(stf_path, config_path)
+        self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
+        self.work_root_path = os.path.join(stf_path, root_path)
+        self.device = device
+        self.template_video_path=template_video_path
+        
+        # model = stf_alternative.create_model(
+        # config_path=config_path,
+        # checkpoint_path=checkpoint_path,
+        # work_root_path=work_root_path,
+        # device=device,
+        # wavlm_path="microsoft/wavlm-large",
+        # )
+        # self.template = stf_alternative.Template(
+        # model=model,
+        # config_path=config_path,
+        # template_video_path=template_video_path,
+        # )
+    
+
+    def execute(self, audio: str):
         
         model = stf_alternative.create_model(
-        config_path=config_path,
-        checkpoint_path=checkpoint_path,
-        work_root_path=work_root_path,
-        device=device,
-        wavlm_path="microsoft/wavlm-large",
+            config_path=self.config_path,
+            checkpoint_path=self.checkpoint_path,
+            work_root_path=self.work_root_path,
+            device=self.device,
+            wavlm_path="microsoft/wavlm-large",
         )
         self.template = stf_alternative.Template(
-        model=model,
-        config_path=config_path,
-        template_video_path=template_video_path,
+            model=model,
+            config_path=self.config_path,
+            template_video_path=self.template_video_path,
         )
-    
 
-    def execute(self, audio: str):
+        
         Path("dubbing").mkdir(exist_ok=True)
         save_path = os.path.join("dubbing", Path(audio).stem+"--lip.mp4")
         reader = iter(self.template._get_reader(num_skip_frames=0))