Update app.py

app.py

@@ -23,7 +23,8 @@ def predict(text):
   text = sbl_normalization(text)
   word_attributions = cls_explainer(text)
   results = pipe(text)[0]
-  label = f"{results['label']} ({results['score']:.2})"
+  label_keys = {'LABEL_0' : 'ABH', 'LABEL_1' : 'CBH', 'LABEL_2' : 'TBH', 'LABEL_3' : 'LBH'}
+  label = f"{label_keys[results['label']]} ({results['score']:.2})"
   return label, word_attributions[1:-1]
 
 iface = gr.Interface(