luigi12345 commited on
Commit
a4f839e
1 Parent(s): 0ad63e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -149,6 +149,7 @@ def main():
149
  model = GlaucomaModel(device=run_device)
150
 
151
  with st.spinner('Analyzing...'):
 
152
  disease_idx, disc_cup_image, vcdr, cls_confidence, cup_confidence, disc_confidence, cropped_image = model.process(image)
153
 
154
  # Display optic disc and cup image
@@ -189,5 +190,4 @@ def main():
189
  st.markdown(final_results_as_table)
190
 
191
  if __name__ == '__main__':
192
- main()
193
-
 
149
  model = GlaucomaModel(device=run_device)
150
 
151
  with st.spinner('Analyzing...'):
152
+ # Get predictions from the model
153
  disease_idx, disc_cup_image, vcdr, cls_confidence, cup_confidence, disc_confidence, cropped_image = model.process(image)
154
 
155
  # Display optic disc and cup image
 
190
  st.markdown(final_results_as_table)
191
 
192
  if __name__ == '__main__':
193
+ main()