Update models/loaders/matanyone_loader.py

models/loaders/matanyone_loader.py

@@ -50,10 +50,10 @@ def _select_device(self, pref: str) -> str:
             return "cpu"
         return "cuda" if torch.cuda.is_available() else "cpu"
     
-    def load(self) -> bool:
+    def load(self):  # <-- CHANGED: No return type hint, returns processor
         """Load MatAnyone using official InferenceCore API."""
         if self.loaded:
-            return True
+            return self.processor  # <-- CHANGED: Return processor, not True
             
         logger.info(f"Loading MatAnyone from HF: {self.model_id} (device={self.device})")
         t0 = time.time()
@@ -69,18 +69,18 @@ def load(self) -> bool:
             self.loaded = True
             self.load_time = time.time() - t0
             logger.info(f"MatAnyone loaded successfully via InferenceCore API in {self.load_time:.2f}s")
-            return True
+            return self.processor  # <-- CHANGED: Return processor, not True
             
         except ImportError as e:
             self.load_error = f"MatAnyone not installed: {e}"
             logger.error(f"Failed to import MatAnyone. Install with: pip install git+https://github.com/pq-yang/MatAnyone.git@main")
-            return False
+            return None  # <-- CHANGED: Return None on failure
             
         except Exception as e:
             self.load_error = str(e)
             logger.error(f"Failed to load MatAnyone: {e}")
             logger.debug(traceback.format_exc())
-            return False
+            return None  # <-- CHANGED: Return None on failure
     
     def process_video(self, video_path: str, mask_path: str, output_dir: Optional[str] = None,
                      max_size: int = 720, save_frames: bool = False) -> Tuple[Optional[str], Optional[str]]: