Update models/loaders/model_loader.py

models/loaders/model_loader.py

@@ -108,7 +108,8 @@ def load_all_models(
                 progress_callback(0.1, "Loading SAM2 model...")
 
             sam2_start = time.time()
-            sam2_model = self.sam2_loader.load()
+            # CHANGED: Force tiny model instead of auto-detection
+            sam2_model = self.sam2_loader.load("tiny")  # Force tiny model for faster loading and less memory usage
             sam2_time = time.time() - sam2_start
 
             if sam2_model:
@@ -306,7 +307,14 @@ def _cleanup_models(self):
         """Internal cleanup of loaded models"""
         # Clean up SAM2
         if self.sam2_loader:
-            self.sam2_loader.cleanup()
+            try:
+                if hasattr(self.sam2_loader, 'cleanup'):
+                    self.sam2_loader.cleanup()
+                else:
+                    logger.debug("SAM2 loader has no cleanup method")
+            except Exception as e:
+                logger.debug(f"SAM2 loader cleanup error: {e}")
+        
         if self.sam2_predictor:
             try:
                 del self.sam2_predictor
@@ -316,7 +324,20 @@ def _cleanup_models(self):
 
         # Clean up MatAnyone
         if self.matanyone_loader:
-            self.matanyone_loader.cleanup()
+            try:
+                if hasattr(self.matanyone_loader, 'cleanup'):
+                    self.matanyone_loader.cleanup()
+                else:
+                    # MatAnyone doesn't have cleanup, but we can clean the wrapper
+                    if hasattr(self.matanyone_loader, '_wrapper') and self.matanyone_loader._wrapper:
+                        if hasattr(self.matanyone_loader._wrapper, 'reset'):
+                            self.matanyone_loader._wrapper.reset()
+                    self.matanyone_loader._processor = None
+                    self.matanyone_loader._wrapper = None
+                    logger.debug("MatAnyone loader cleaned up manually")
+            except Exception as e:
+                logger.debug(f"MatAnyone loader cleanup error: {e}")
+        
         if self.matanyone_model:
             try:
                 del self.matanyone_model
@@ -331,4 +352,4 @@ def _cleanup_models(self):
         # Garbage collection
         gc.collect()
 
-        logger.debug("Model cleanup completed")
+        logger.debug("Model cleanup completed")