Update pipeline/video_pipeline.py
Browse files
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")
|