Update tasks/text.py
Browse files- 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 |
|