change import

tasks/text.py

@@ -4,6 +4,7 @@ from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import random
 from skops.io import load
+from .utils.text_preprocessor import TextPreprocessor
 
 
 from .utils.evaluation import TextEvaluationRequest
@@ -28,15 +29,14 @@ def baseline_model(dataset_length: int):
     return predictions
 
 def tree_classifier(test_dataset: dict, model: str):
-    from .utils.text_preprocessor import TextPreprocessor
-    
+
     texts = test_dataset["quote"]
 
     model_path = f"tasks/text_models/{model}.skops"
 
     model = load(model_path, 
                  trusted=[
-                     '__main__.TextPreprocessor',
+                     'utils.text_preprocessor.TextPreprocessor',
                      'nltk.stem.wordnet.WordNetLemmatizer',
                      'xgboost.core.Booster',
                      'xgboost.sklearn.XGBClassifier'])