Update app.py

app.py

@@ -232,7 +232,8 @@ 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(tiles, conf=0.5) # predict     
+            st.rerun()
         
         # plot predicted image
         im_bgr = st.session_state.results[st.session_state.image_counter].plot()  # BGR-order numpy array