Create models.py

models.py

@@ -116,8 +116,8 @@ class ModelManager:
                     try:
                         logger.info("Loading custom intent model with MultiLabelIntentClassifier")
                         
-                        # Create the custom model architecture
-                        self.models[name] = MultiLabelIntentClassifier("distilbert-base-uncased", 6)
+                        # Create the custom model architecture (referencing the global class)
+                        intent_model = MultiLabelIntentClassifier("distilbert-base-uncased", 6)
                         
                         # Download the model file from HuggingFace
                         try:
@@ -128,8 +128,8 @@ class ModelManager:
                             
                             # Load the state dict
                             state_dict = torch.load(model_path, map_location='cpu')
-                            self.models[name].load_state_dict(state_dict)
-                            self.models[name] = self.models[name].to(self.device)
+                            intent_model.load_state_dict(state_dict)
+                            self.models[name] = intent_model.to(self.device)
                             
                             logger.info("Custom intent model loaded successfully from HuggingFace")