elucidator8918 commited on
Commit
c2d7e1a
·
verified ·
1 Parent(s): 1f5243a

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +2 -0
tasks/text.py CHANGED
@@ -3,6 +3,8 @@ from datetime import datetime
3
  from datasets import load_dataset
4
  from sklearn.metrics import accuracy_score
5
  import random
 
 
6
 
7
  from .utils.evaluation import TextEvaluationRequest
8
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
3
  from datasets import load_dataset
4
  from sklearn.metrics import accuracy_score
5
  import random
6
+ from transformers import AutoConfig, AutoModelForSequenceClassification, AutoTokenizer
7
+ import torch
8
 
9
  from .utils.evaluation import TextEvaluationRequest
10
  from .utils.emissions import tracker, clean_emissions_data, get_space_info