Update stf_utils.py

stf_utils.py

@@ -134,15 +134,15 @@ class STFPipeline:
 
         
         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
+            # 순차적으로 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