seronk commited on
Commit
4342aae
·
verified ·
1 Parent(s): 30062ab

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -0
tasks/text.py CHANGED
@@ -10,6 +10,7 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
10
  #additional imports
11
  from transformers import Trainer, TrainingArguments, DistilBertForSequenceClassification, DistilBertTokenizerFast, AutoModelForSequenceClassification,DataCollatorWithPadding, AutoTokenizer
12
  import logging
 
13
 
14
  router = APIRouter()
15
 
 
10
  #additional imports
11
  from transformers import Trainer, TrainingArguments, DistilBertForSequenceClassification, DistilBertTokenizerFast, AutoModelForSequenceClassification,DataCollatorWithPadding, AutoTokenizer
12
  import logging
13
+ import numpy as np
14
 
15
  router = APIRouter()
16