Update app.py

app.py

@@ -40,7 +40,7 @@ if text:
     title, probabilities, highest = predict_sentence(model, tokenizer, text)
     out = {
         'title': title,
-        'prob': probabilities
+        'prob': highest
     }
     st.json(out)