MogensR commited on
Commit
d4c672c
·
verified ·
1 Parent(s): 7a95b50

Update pipeline/video_pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline/video_pipeline.py +1 -1
pipeline/video_pipeline.py CHANGED
@@ -215,7 +215,7 @@ def stage1_create_transparent_video(input_file):
215
  threading.Thread(target=heartbeat_monitor, args=(heartbeat_flag,), daemon=True).start()
216
  try:
217
  # Load models
218
- sam2_predictor = load_sam2_predictor()
219
  matanyone_processor = load_matanyone_processor()
220
  if not sam2_predictor or not matanyone_processor:
221
  raise RuntimeError("Failed to load models")
 
215
  threading.Thread(target=heartbeat_monitor, args=(heartbeat_flag,), daemon=True).start()
216
  try:
217
  # Load models
218
+ sam2_predictor, _ = load_sam2_predictor() # Extract predictor from tuple
219
  matanyone_processor = load_matanyone_processor()
220
  if not sam2_predictor or not matanyone_processor:
221
  raise RuntimeError("Failed to load models")