Update app.py

app.py

@@ -203,7 +203,7 @@ def tile_image(image, tile_size, overlap):
         
 if st.session_state.screen == 2:  
     # tile images
-    tiles = tile_image(image=st.session_state.image, tile_size=640, overlap=40)
+    st.session_state.tiles = tile_image(image=st.session_state.image, tile_size=640, overlap=40)
 
     with row5_col2:
         st.image(st.session_state.image, use_column_width=True)
@@ -229,7 +229,7 @@ if st.session_state.screen == 2:
         if 'results' not in st.session_state:
             st.write("The model is working. Please be patient...")
             st.write("This process takes around 30 seconds.")
-            st.session_state.results = st.session_state.model.predict(tiles, conf=0.5) # predict     
+            st.session_state.results = st.session_state.model.predict(st.session_state.tiles[0], conf=0.5) # predict     
             st.rerun()
         
         # plot predicted image