Update app.py

app.py

@@ -30,11 +30,12 @@ def main():
     # Instructions
     st.markdown("*Hint: click on the top-right corner of an image to enlarge it!*")
     # Set the columns
-    cols = st.beta_columns((1, 1))
+    cols = st.beta_columns((1, 1, 1))
     cols[0].subheader("Input image")
-    cols[1].subheader("Class activation map")
+    cols[1].subheader("Optic disc and optic cup")
+    cols[2].subheader("Class activation map")
     
-    # Set the visualization figure
+    # set the visualization figure
     fig, ax = plt.subplots()
 
     # Sidebar
@@ -45,9 +46,10 @@ def main():
     # Choose your own image
     uploaded_file = st.sidebar.file_uploader("Upload image", type=['png', 'jpeg', 'jpg'])
     if uploaded_file is not None:
-        # Read the upload image
+        # read the upload image
         image = Image.open(uploaded_file).convert('RGB')
         image = np.array(image).astype(np.uint8)
+        # page_idx = 0
         ax.imshow(image)
         ax.axis('off')
         cols[0].pyplot(fig)
@@ -55,7 +57,7 @@ def main():
     # For newline
     st.sidebar.write('\n')
 
-    # Actions
+    # actions
     if st.sidebar.button("Analyze image"):
 
         if uploaded_file is None:
@@ -63,22 +65,34 @@ def main():
 
         else:
             with st.spinner('Loading model...'):
-                # Load model
+                # load model
                 model = GlaucomaModel(device=run_device)
 
             with st.spinner('Analyzing...'):
                 # Forward the image to the model and get results
-                disease_idx, cam = model.process(image)
+                disease_idx, disc_cup_image, cam, vcdr = model.process(image)
 
-                # Plot the class activation map
-                ax.imshow(cam)
+                # plot the optic disc and optic cup image
+                ax.imshow(disc_cup_image)
                 ax.axis('off')
                 cols[1].pyplot(fig)
 
-                # Display screening results
+                # plot the stitched image
+                ax.imshow(cam)
+                ax.axis('off')
+                cols[2].pyplot(fig)
+
+                # Display JSON
                 st.subheader("  Screening results:")
                 st.write('\n')
-                st.markdown(f"{model.id2label[disease_idx]}")
+                
+                final_results_as_table = f"""
+                |Params|Outcomes|
+                |---|---|
+                |vCDR|{vcdr:.04f}|
+                |Category|{model.cls_id2label[disease_idx]}|
+                """
+                st.markdown(final_results_as_table)
 
 
 if __name__ == '__main__':