Update app.py

app.py

@@ -34,7 +34,7 @@ def main():
     cols[0].subheader("Input image")
     cols[1].subheader("Class activation map")
     
-    # set the visualization figure
+    # Set the visualization figure
     fig, ax = plt.subplots()
 
     # Sidebar
@@ -45,10 +45,9 @@ 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)
@@ -56,7 +55,7 @@ def main():
     # For newline
     st.sidebar.write('\n')
 
-    # actions
+    # Actions
     if st.sidebar.button("Analyze image"):
 
         if uploaded_file is None:
@@ -64,22 +63,19 @@ 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)
-                
-                # visualize results
-                # fig, ax = plt.subplots()
 
-                # plot the stitched image
+                # Plot the class activation map
                 ax.imshow(cam)
                 ax.axis('off')
                 cols[1].pyplot(fig)
 
-                # Display JSON
+                # Display screening results
                 st.subheader("  Screening results:")
                 st.write('\n')
                 st.markdown(f"{model.id2label[disease_idx]}")