lalithadevi
commited on
Commit
•
9e645b6
1
Parent(s):
f97e412
Update news_category_prediction.py
Browse files
news_category_prediction.py
CHANGED
@@ -19,6 +19,7 @@ def parse_prediction(tflite_pred, label_encoder):
|
|
19 |
def inference(text, interpreter, label_encoder, tokenizer):
|
20 |
logger.warning('Entering inference()')
|
21 |
batch_size = len(text)
|
|
|
22 |
if text != "":
|
23 |
tokens = tokenizer(text, max_length=DISTILBERT_TOKENIZER_N_TOKENS, padding="max_length", truncation=True, return_tensors="tf")
|
24 |
# tflite model inference
|
|
|
19 |
def inference(text, interpreter, label_encoder, tokenizer):
|
20 |
logger.warning('Entering inference()')
|
21 |
batch_size = len(text)
|
22 |
+
logger.warning(f'Samples to predict: {batch_size}')
|
23 |
if text != "":
|
24 |
tokens = tokenizer(text, max_length=DISTILBERT_TOKENIZER_N_TOKENS, padding="max_length", truncation=True, return_tensors="tf")
|
25 |
# tflite model inference
|