Update app.py

app.py

@@ -99,7 +99,18 @@ class DepartmentClassifierService:
             "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
         )
 
-        self.classifier = joblib.load(model_path)
+        model_package = joblib.load(model_path)
+
+        if isinstance(model_package, dict):
+            self.classifier = model_package["classifier"]
+            self.model_classes = model_package.get("classes", list(self.classifier.classes_))
+            self.embedding_model_name = model_package.get(
+                "embedding_model_name",
+                self.embedding_model_name
+            )
+        else:
+            self.classifier = model_package
+            self.model_classes = list(self.classifier.classes_)
 
         self.embedding_model = SentenceTransformer(self.embedding_model_name)