Update app.py

app.py

@@ -119,19 +119,19 @@ def main():
                     download_confident_images.append((cropped_image, uploaded_file.name))
                 
                 # Display Results
-                with st.container():
-                    st.subheader(f"Results for {uploaded_file.name}")
-                    cols = st.beta_columns(4)  # Use st.beta_columns for compatibility with older Streamlit
-                    cols[0].image(image_np, caption="Input Image", use_column_width=True)
-                    cols[1].image(disc_cup_image, caption="Disc/Cup Segmentation", use_column_width=True)
-                    cols[2].image(image_np, caption="Class Activation Map", use_column_width=True)
-                    cols[3].image(cropped_image, caption="Cropped Image", use_column_width=True)
+                st.subheader(f"Results for {uploaded_file.name}")
+                cols = st.beta_columns(4)  # Use st.beta_columns for compatibility with older Streamlit
+                cols[0].image(image_np, caption="Input Image", use_column_width=True)
+                cols[1].image(disc_cup_image, caption="Disc/Cup Segmentation", use_column_width=True)
+                cols[2].image(image_np, caption="Class Activation Map", use_column_width=True)
+                cols[3].image(cropped_image, caption="Cropped Image", use_column_width=True)
 
-                    st.write(f"**Vertical cup-to-disc ratio:** {vcdr:.04f}")
-                    st.write(f"**Category:** {model.cls_id2label[disease_idx]} ({cls_conf:.02f}% confidence)")
-                    st.write(f"**Optic Cup Segmentation Confidence:** {cup_conf:.02f}%")
-                    st.write(f"**Optic Disc Segmentation Confidence:** {disc_conf:.02f}%")
-                    st.write(f"**Confidence Group:** {'Confident' if is_confident else 'Not Confident'}")
+                # Display confidence and metrics
+                st.write(f"**Vertical cup-to-disc ratio:** {vcdr:.04f}")
+                st.write(f"**Category:** {model.cls_id2label[disease_idx]} ({cls_conf:.02f}% confidence)")
+                st.write(f"**Optic Cup Segmentation Confidence:** {cup_conf:.02f}%")
+                st.write(f"**Optic Disc Segmentation Confidence:** {disc_conf:.02f}%")
+                st.write(f"**Confidence Group:** {'Confident' if is_confident else 'Not Confident'}")
         
         # Download Link for Confident Images
         if download_confident_images: