charting added

app.py

@@ -3,6 +3,7 @@ import torch
 from torch import nn
 from transformers import BertModel, AutoTokenizer, AutoModel, pipeline
 from time import time
+import matplotlib.pyplot as plt
 # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 device = 'cpu'
 
@@ -72,4 +73,15 @@ for lbl, score in res:
     if score >=1:
         st.markdown(f"[        {lbl:<7}] {labels_decoder[lbl]:<35}  {score:.1f}%")
 
+
+fig, ax = plt.subplots()
+total=0
+for r in res:
+    if total < 95:
+        ax.barh(r[0], r[1])
+        total += r[1]
+    else:
+        break
+st.pyplot()
+
 st.markdown(f"cycle time = {time() - start_time:.2f} s.")