dhhd255 commited on
Commit
7b69f6b
·
1 Parent(s): 0a3512b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- st.caption(f'{confidence*100:.0f}% sure')
81
  else:
82
  col2.markdown('<span class="result healthy">Predicted class: Healthy</span>', unsafe_allow_html=True)
83
- st.caption(f'{confidence*100:.0f}% sure')
 
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)