Update app.py

app.py

@@ -15,7 +15,7 @@ def print_probs(logits):
   i = 0
   while sum <= 95:
     prob, idx = ans[i]
-    text = categories[idx] + ": "+ str(np.round(prob,1))
+    text = categories[idx] + ": "+ str(np.round(prob,1)) + "%"
     st.markdown(text)
     sum+=prob
     i+=1
@@ -24,7 +24,7 @@ def make_prediction(text):
   tokenized_text = tokenizer(text, return_tensors='pt')
   with torch.no_grad():
     pred_logits = model(**tokenized_text).logits
-  st.markdown("Predictions:")
+  st.markdown("Category probability:")
   print_probs(pred_logits[0])