Spaces:

YANGYYYY
/

cartoonize

Running

YANGYYYY commited on May 26

Commit

e7d45cf

•

1 Parent(s): ca106e1

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -267,7 +267,7 @@ class Predictor:
                 frames[frame_idx] = frame
                 frame_idx += 1
                 if frame_idx == batch_size:
-                    transformed_frames = self.transform_and_save(frames, frame_idx)
                     transformed_video_frames.extend(transformed_frames)
                     frame_idx = 0
         except Exception as e:

                 frames[frame_idx] = frame
                 frame_idx += 1
                 if frame_idx == batch_size:
+                    transformed_frames = transform_and_save(frames, frame_idx)
                     transformed_video_frames.extend(transformed_frames)
                     frame_idx = 0
         except Exception as e: