Update app.py

app.py

@@ -13,16 +13,20 @@ import os
 
 app = Flask(__name__)
 CORS(app)
-logging.warning('Initiated')
+FORMAT = '%(asctime)s %(message)s'
+logging.basicConfig(format=FORMAT)
+logger = logging.getLogger('hf_logger')
 
 
 def load_model():
+    logger.warning('Entering load transformer')
     interpreter = tf.lite.Interpreter(model_path=os.path.join("models/news_classification_hf_distilbert.tflite"))
     with open("models/news_classification_labelencoder.bin", "rb") as model_file_obj:
         label_encoder = cloudpickle.load(model_file_obj)
         
     model_checkpoint = "distilbert-base-uncased"
     tokenizer = DistilBertTokenizerFast.from_pretrained(model_checkpoint)
+    logger.warning('Exiting load transformer')
     return interpreter, label_encoder, tokenizer
 
 interpreter, label_encoder, tokenizer = load_model()
@@ -31,6 +35,7 @@ interpreter, label_encoder, tokenizer = load_model()
 @app.route("/")
 @cross_origin()
 def update_news():
+    logger.warning('Entering update_news()')
     status_json = "{'status':'success'}"
     status_code = 200
     try:
@@ -44,6 +49,7 @@ def update_news():
     except:
         status_json = "{'status':'failure'}"
         status_code = 500
+    logger.warning('Exiting update_news()')
     return Response(status_json, status=status_code, mimetype='application/json')