Update app.py

app.py

@@ -30,13 +30,20 @@ if 'image_counter' not in st.session_state:
 #####
 # This displays while the model is loading
 # It provides the user with information about the tool
+# This process takes around 30 seconds
 _, row0_col1, _ = st.columns([1,3,1])
-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: center; 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 option), this tool find Waldo and his friends Wenda, Odlaw, Wizard, and Woof. \
-    The model takes about 30 seconds to load before the fun begins. Please be patient...</p>", unsafe_allow_html=True)
+if 
+# load model
+if 'model' not in st.session_state:
+    with row0_col1:
+        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: center; 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 option), this tool find Waldo and his friends Wenda, Odlaw, Wizard, and Woof. \
+            The model takes about 30 seconds to load before the fun begins. Please be patient...</p>", unsafe_allow_html=True)
+        st.image()
+    st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load    
     
 #####
 ###
@@ -119,12 +126,7 @@ with row3_col5:
     img_8 = ImageOps.exif_transpose(img_8)
     st.image(img_8, use_column_width=True)
     if st.button(f"Select Image ↑", key="button_8"):
-        st.session_state.image = img_8
-
-# load model
-if 'model' not in st.session_state:
-    st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load        
-    
+        st.session_state.image = img_8   
 
 #####
 ###