luigi12345
commited on
Commit
•
a4f839e
1
Parent(s):
0ad63e5
Update app.py
Browse files
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()
|
|