laureBe commited on
Commit
24930f7
·
verified ·
1 Parent(s): 721a035

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +5 -2
tasks/text.py CHANGED
@@ -10,12 +10,15 @@ import tensorflow as tf
10
  from transformers import DistilBertTokenizer
11
  from transformers import TFDistilBertForSequenceClassification
12
  from transformers import logging
13
- logging.set_verbosity_error()
14
- logging.set_verbosity_warning()
15
  from .utils.evaluation import TextEvaluationRequest
16
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
17
  import os
 
18
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'
 
 
 
 
19
 
20
  router = APIRouter()
21
 
 
10
  from transformers import DistilBertTokenizer
11
  from transformers import TFDistilBertForSequenceClassification
12
  from transformers import logging
 
 
13
  from .utils.evaluation import TextEvaluationRequest
14
  from .utils.emissions import tracker, clean_emissions_data, get_space_info
15
  import os
16
+ import re
17
  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '1'
18
+ os.environ['TF_ENABLE_ONEDNN_OPTS'] ='0'
19
+ logging.set_verbosity_error()
20
+ logging.set_verbosity_warning()
21
+
22
 
23
  router = APIRouter()
24