From 3a59e87f3e4db0a5ebaa28280807427dc4c13cd4 Mon Sep 17 00:00:00 2001
From: Scott Register <sreg@sreg.io>
Date: Sun, 27 Jul 2025 08:58:43 -0700
Subject: [PATCH] fix something

---
 vr180_streaming/sam2_streaming.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vr180_streaming/sam2_streaming.py b/vr180_streaming/sam2_streaming.py
index ccb16a4..2f06f95 100644
--- a/vr180_streaming/sam2_streaming.py
+++ b/vr180_streaming/sam2_streaming.py
@@ -144,7 +144,7 @@ class SAM2StreamingProcessor:
                 'temp_output_dict_per_obj': {},
                 'consolidated_frame_inds': {},
                 'tracking_has_started': False,
-                'num_frames': video_info['frame_count'],
+                'num_frames': video_info.get('total_frames', video_info.get('frame_count', 0)),
                 'video_height': video_info['height'],
                 'video_width': video_info['width'],
                 'device': self.device,