Spaces:
Running
Running
Update tasks/text.py
Browse files- 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 =
|
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 |
|