Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -77,9 +77,10 @@ if uploaded_file is not None:
|
|
77 |
confidence = torch.softmax(logits, dim=1)[0][predicted_class].item()
|
78 |
if(predicted_class == 0):
|
79 |
col2.markdown('<span class="result parkinsons">Predicted class: Parkinson\'s</span>', unsafe_allow_html=True)
|
80 |
-
|
81 |
else:
|
82 |
col2.markdown('<span class="result healthy">Predicted class: Healthy</span>', unsafe_allow_html=True)
|
83 |
-
|
|
|
84 |
# Add the footer
|
85 |
st.markdown('<div class="footer">Hello</div>', unsafe_allow_html=True)
|
|
|
77 |
confidence = torch.softmax(logits, dim=1)[0][predicted_class].item()
|
78 |
if(predicted_class == 0):
|
79 |
col2.markdown('<span class="result parkinsons">Predicted class: Parkinson\'s</span>', unsafe_allow_html=True)
|
80 |
+
col2.caption(f'{confidence*100:.0f}% sure')
|
81 |
else:
|
82 |
col2.markdown('<span class="result healthy">Predicted class: Healthy</span>', unsafe_allow_html=True)
|
83 |
+
col2.caption(f'{confidence*100:.0f}% sure')
|
84 |
+
|
85 |
# Add the footer
|
86 |
st.markdown('<div class="footer">Hello</div>', unsafe_allow_html=True)
|