remove print

app.py

@@ -499,8 +499,6 @@ def propagate_masks(GLOBAL_STATE: gr.State):
     with torch.inference_mode():
         for frame_idx, frame in enumerate(GLOBAL_STATE.video_frames):
             pixel_values = None
-            print("inference_session.processed_frames", inference_session.processed_frames)
-            print("frame_idx", frame_idx)
             if inference_session.processed_frames is None or frame_idx not in inference_session.processed_frames:
                 pixel_values = processor(images=frame, device="cuda", return_tensors="pt").pixel_values[0]
             sam2_video_output = model(inference_session=inference_session, frame=pixel_values, frame_idx=frame_idx)
@@ -754,32 +752,17 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation", the
         out_path = "/tmp/sam2_playback.mp4"
         # Prefer imageio with PyAV/ffmpeg to respect exact fps
         try:
-            import imageio.v3 as iio  # type: ignore
+            import cv2  # type: ignore
 
-            iio.imwrite(out_path, [fr[:, :, ::-1] for fr in frames_np], plugin="pyav", fps=fps)
+            fourcc = cv2.VideoWriter_fourcc(*"mp4v")
+            writer = cv2.VideoWriter(out_path, fourcc, fps, (w, h))
+            for fr_bgr in frames_np:
+                writer.write(fr_bgr)
+            writer.release()
             return out_path
         except Exception as e:
-            print(f"Failed to render video with imageio.v3: {e}")
-            # Fallbacks
-            try:
-                import imageio.v2 as imageio  # type: ignore
-
-                imageio.mimsave(out_path, [fr[:, :, ::-1] for fr in frames_np], fps=fps)
-                return out_path
-            except Exception as e:
-                print(f"Failed to render video with imageio.v2: {e}")
-                try:
-                    import cv2  # type: ignore
-
-                    fourcc = cv2.VideoWriter_fourcc(*"mp4v")
-                    writer = cv2.VideoWriter(out_path, fourcc, fps, (w, h))
-                    for fr_bgr in frames_np:
-                        writer.write(fr_bgr)
-                    writer.release()
-                    return out_path
-                except Exception as e:
-                    print(f"Failed to render video with cv2: {e}")
-                    raise gr.Error(f"Failed to render video: {e}")
+            print(f"Failed to render video with cv2: {e}")
+            raise gr.Error(f"Failed to render video: {e}")
 
     render_btn.click(_render_video, inputs=[GLOBAL_STATE], outputs=[playback_video])