csk99 commited on
Commit
31bc5e0
·
verified ·
1 Parent(s): 3034e17

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -1
tasks/text.py CHANGED
@@ -75,7 +75,7 @@ async def evaluate_text(request: TextEvaluationRequest):
75
  xgb_multi = pickle.load(f)
76
 
77
 
78
- X_train = test_dataset["quote"].copy()
79
 
80
  y_train = test_dataset["label"].copy()
81
 
 
75
  xgb_multi = pickle.load(f)
76
 
77
 
78
+ X_train = sentence_embeddings.copy()
79
 
80
  y_train = test_dataset["label"].copy()
81