Update stf_utils.py

stf_utils.py

@@ -130,22 +130,41 @@ class STFPipeline:
         audio_segment = AudioSegment.from_file(audio)
         pivot = 0
         results = []
-        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
+
+
+        
+        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
 
         print('STFPipeline execute 3')
         images2video(results, save_path)