Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -72,7 +72,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
72 |
model = SentenceTransformer("dunzhang/stella_en_400M_v5",trust_remote_code=True).cuda()
|
73 |
|
74 |
trusted_types = ['sklearn.feature_selection._univariate_selection.f_classif']
|
75 |
-
disp = sio.load('logistic_regression_model.skops',trusted=trusted_types)
|
76 |
|
77 |
## Data prep
|
78 |
embeddings = model.encode(list(test_dataset['quote']), prompt_name=query_prompt_name)
|
|
|
72 |
model = SentenceTransformer("dunzhang/stella_en_400M_v5",trust_remote_code=True).cuda()
|
73 |
|
74 |
trusted_types = ['sklearn.feature_selection._univariate_selection.f_classif']
|
75 |
+
disp = sio.load('./tasks/logistic_regression_model.skops',trusted=trusted_types)
|
76 |
|
77 |
## Data prep
|
78 |
embeddings = model.encode(list(test_dataset['quote']), prompt_name=query_prompt_name)
|