Update app.py

app.py

@@ -167,7 +167,7 @@ os.system("rm -R -- */")
 # os.system("mkdir -p predictions")
 
 with col:
-    with st.container:
+    with st.container():
         # Create heading and description
         st.markdown("<h1 align='center'>Cavity Detection Tool</h1>", unsafe_allow_html=True)    
         st.markdown("Cavity Detection Tool (CADET) is a machine learning pipeline trained to detect X-ray cavities from noisy Chandra images of early-type galaxies.")