Oriaz commited on
Commit
a2de25b
·
verified ·
1 Parent(s): 694d846

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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)