Update app.py

app.py

@@ -14,8 +14,8 @@ def print_probs(logits):
   i = 0
   while sum <= 95:
     prob, idx = ans[i]
-    text = categories[idx] + ": "+ str(np.round(prob,1)
-    # print(text)
+    text = categories[idx] + ": "+ str(np.round(prob,1))
+    st.markdown(text)
     sum+=prob
     i+=1
 
@@ -31,7 +31,7 @@ def make_prediction(text):
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
 model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
-model_name = train_model2
+model_name = "train_model2"
 model_path = model_name + '.zip'
 model.load_state_dict(
     torch.load(