Update stf_utils.py

stf_utils.py

@@ -132,39 +132,23 @@ class STFPipeline:
         results = []
 
 
-        
-        try:
-            # 순차적으로 gen_infer를 실행
-            gen_infer = self.template.gen_infer(audio_segment, pivot)
     
-            for idx, (it, chunk) in enumerate(gen_infer, pivot):
-                frame = next(reader)  # 다음 프레임을 읽음
-                composed = self.template.compose(idx, frame, it)  # 합성 작업 수행
-                frame_name = f"{idx}".zfill(5) + ".jpg"
-                results.append(it['pred'])  # 결과를 저장
-            pivot = idx + 1
-
-        except StopIteration as e:
-            pass
-
-
-        
-        # with ThreadPoolExecutor(4) as p:
-        #     try:
-
-        #         gen_infer = self.template.gen_infer_concurrent(
-        #             p,
-        #             audio_segment,
-        #             pivot,
-        #         )
-        #         for idx, (it, chunk) in enumerate(gen_infer, pivot):
-        #             frame = next(reader)
-        #             composed = self.template.compose(idx, frame, it)
-        #             frame_name = f"{idx}".zfill(5)+".jpg"
-        #             results.append(it['pred'])
-        #         pivot = idx + 1
-        #     except StopIteration as e:
-        #         pass
+        with ThreadPoolExecutor(4) as p:
+            try:
+
+                gen_infer = self.template.gen_infer_concurrent(
+                    p,
+                    audio_segment,
+                    pivot,
+                )
+                for idx, (it, chunk) in enumerate(gen_infer, pivot):
+                    frame = next(reader)
+                    composed = self.template.compose(idx, frame, it)
+                    frame_name = f"{idx}".zfill(5)+".jpg"
+                    results.append(it['pred'])
+                pivot = idx + 1
+            except StopIteration as e:
+                pass
 
         print('STFPipeline execute 3')
         images2video(results, save_path)