submission-template

Runtime error

kantundpeterpan commited on Jan 18

Commit

1c1c1db

1 Parent(s): 931ac30

ready1234

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -102,7 +102,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     # Make predictions
     true_labels = test_dataset["label"]
     predictions = [
-        LABEL_MAPPING[r] for r in model.predict(test_dataset)
     ]
     #--------------------------------------------------------------------------------------------

     # Make predictions
     true_labels = test_dataset["label"]
     predictions = [
+        LABEL_MAPPING[r] for r in model.predict(test_df)
     ]
     #--------------------------------------------------------------------------------------------