Update app.py

app.py

@@ -23,16 +23,7 @@ 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.warning(f"{model_res = }")
-    logger.warning(f"{model_res['label'] = }")
-    print(f"{model_res['label'] = }")
-    # label_map = {"LABEL_0": "NOT RISKY", "LABEL_1": "RISKY"}
-    label_map = {
-    "0": "Not Risky",
-    "1": "Possibly Risky",
-    "2": "Risky"
-  }
-    label_res = label_map.get(model_res["label"])
+    label_res = model_res["label"]
     score = model_res["score"]
     res = f"Result: {label_res}\n\nScore: {score}"
     return res