fix: Add logging to figure out label mapping

app.py

@@ -3,6 +3,9 @@ from transformers import DistilBertTokenizer
 from transformers import DistilBertForSequenceClassification
 from transformers import pipeline
 import gradio as gr
+import logging
+
+logger = logging.getLogger(__name__)
 
 MODEL_PATH = "RedmarkerAI/hrw_multi_generous_v1"
 auth_token = os.environ.get("TOKEN_MODEL")
@@ -20,6 +23,8 @@ def clf_result(text_input: str) -> str:
         res = "Please enter a sentence with the word `best`"
         return res
     model_res = clf(text_input)[0]
+    logger.info(f"{model_res = }")
+    logger.info(f"{model_res['label'] = }")
     # label_map = {"LABEL_0": "NOT RISKY", "LABEL_1": "RISKY"}
     label_map = {
     "0": "Not Risky",