Update app.py

app.py

@@ -74,10 +74,12 @@ if uploaded_file is not None:
 
         logits = feature_reducer(logits)
         predicted_class = torch.argmax(logits, dim=1).item()
+        confidence = torch.softmax(logits, dim=1)[0][predicted_class].item()
         if(predicted_class == 0):
             col2.markdown('<span class="result parkinsons">Predicted class: Parkinson\'s</span>', unsafe_allow_html=True)
+            st.caption(f'{confidence*100:.0f}% sure')
         else:
             col2.markdown('<span class="result healthy">Predicted class: Healthy</span>', unsafe_allow_html=True)
-
+            st.caption(f'{confidence*100:.0f}% sure')
 # Add the footer
 st.markdown('<div class="footer">Hello</div>', unsafe_allow_html=True)