Update app.py
Browse files
app.py
CHANGED
|
@@ -122,12 +122,12 @@ def load_models(self, progress_callback: Optional[Callable] = None) -> str:
|
|
| 122 |
if self.cancel_event.is_set():
|
| 123 |
return "Model loading cancelled"
|
| 124 |
|
| 125 |
-
# Initialize core processor with loaded models
|
| 126 |
self.core_processor = video_processor.CoreVideoProcessor(
|
| 127 |
sam2_predictor=sam2_predictor,
|
| 128 |
matanyone_model=matanyone_model,
|
| 129 |
config=self.config,
|
| 130 |
-
|
| 131 |
)
|
| 132 |
|
| 133 |
# Initialize two-stage processor if available
|
|
|
|
| 122 |
if self.cancel_event.is_set():
|
| 123 |
return "Model loading cancelled"
|
| 124 |
|
| 125 |
+
# Initialize core processor with loaded models - FIXED parameter name
|
| 126 |
self.core_processor = video_processor.CoreVideoProcessor(
|
| 127 |
sam2_predictor=sam2_predictor,
|
| 128 |
matanyone_model=matanyone_model,
|
| 129 |
config=self.config,
|
| 130 |
+
memory_mgr=self.memory_manager
|
| 131 |
)
|
| 132 |
|
| 133 |
# Initialize two-stage processor if available
|