Update stf_utils.py

stf_utils.py

@@ -96,9 +96,13 @@ class STFPipeline:
         # config_path=config_path,
         # template_video_path=template_video_path,
         # )
+
+        print('STFPipeline init')
     
 
     def execute(self, audio: str):
+
+        print('STFPipeline execute')
         
         model = stf_alternative.create_model(
             config_path=self.config_path,
@@ -107,12 +111,15 @@ class STFPipeline:
             device=self.device,
             wavlm_path="microsoft/wavlm-large",
         )
+
+        print('STFPipeline execute 1')
         self.template = stf_alternative.Template(
             model=model,
             config_path=self.config_path,
             template_video_path=self.template_video_path,
         )
 
+        print('STFPipeline execute 2')
         
         # Path("dubbing").mkdir(exist_ok=True)
         # save_path = os.path.join("dubbing", Path(audio).stem+"--lip.mp4")
@@ -139,7 +146,8 @@ class STFPipeline:
                 pivot = idx + 1
             except StopIteration as e:
                 pass
-            
+
+        print('STFPipeline execute 3')
         images2video(results, save_path)
                                 
         return save_path