Update news_category_prediction.py

news_category_prediction.py

@@ -19,6 +19,7 @@ def parse_prediction(tflite_pred, label_encoder):
 def inference(text, interpreter, label_encoder, tokenizer):
     logger.warning('Entering inference()')
     batch_size = len(text)
+    logger.warning(f'Samples to predict: {batch_size}')
     if text != "":
         tokens = tokenizer(text, max_length=DISTILBERT_TOKENIZER_N_TOKENS, padding="max_length", truncation=True, return_tensors="tf")
         # tflite model inference