Update tasks/text.py

tasks/text.py

@@ -4,7 +4,7 @@ from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 
 from skops import hub_utils
-from skops.io import load
+import skops.io as sio
 
 from huggingface_hub import hf_hub_download
 import joblib
@@ -87,8 +87,9 @@ async def evaluate_text(request: TextEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
 
     #load model
-    model = joblib.load(
-        hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+    model = sio.load(
+        hf_hub_download(repo_id=REPO_ID, filename=FILENAME),
+        trusted=['__main__.lemmatize_X', '__main__.tokenize_quote']
     )
     
     # Make predictions