Update app.py

app.py

@@ -43,7 +43,7 @@ if 'image_counter' not in st.session_state:
 if 'model' not in st.session_state:
     st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load 
 
-_, row0_col1, _ = st.columns([1,3,1])
+_, row0_col1, _ = st.columns([2,3,2])
 _, row1_col1, _ = st.columns([2,3,2])
 
 
@@ -63,7 +63,6 @@ if st.session_state.screen == 0:
         st.markdown("<h1 style='text-align: center; color: black;'>Find Waldo and His Friends</h1>", unsafe_allow_html=True)
         st.markdown("<h3 style='text-align: center; color: black;'>Using Computer Vision</h3>", unsafe_allow_html=True)
         st.markdown("<p style='text-align: left; color: black;'>This tool allows you to take a photo with your phone of a <i>Where's Waldo?</i> book page. Using the computer vision model YOLOv8-large, this tool find Waldo and his friends Wenda, Odlaw, Wizard, and Woof.</p>", unsafe_allow_html=True)
-        st.markdown("<p style='text-align: left; color: black;'>Once the model has loaded, this screen will automatically change.</p>", unsafe_allow_html=True)
         if st.button(f"Proceed", key="Proceed"):
             st.session_state.screen = 1 
             st.rerun()