Rcarvalo commited on
Commit
c6e64ec
·
verified ·
1 Parent(s): 90766c3

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -0
tasks/text.py CHANGED
@@ -7,6 +7,7 @@ import torch
7
  from .utils.evaluation import TextEvaluationRequest
8
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
9
  from transformers import AutoTokenizer,BertForSequenceClassification,AutoModelForSequenceClassification,Trainer, TrainingArguments,DataCollatorWithPadding
 
10
 
11
  router = APIRouter()
12
 
 
7
  from .utils.evaluation import TextEvaluationRequest
8
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
9
  from transformers import AutoTokenizer,BertForSequenceClassification,AutoModelForSequenceClassification,Trainer, TrainingArguments,DataCollatorWithPadding
10
+ import numpy as np
11
 
12
  router = APIRouter()
13