Update app.py

app.py

@@ -108,6 +108,7 @@ with row3_col5:
 # load model
 if 'model' not in st.session_state:
     st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load        
+    
 
 #####
 ###
@@ -139,7 +140,8 @@ if st.session_state.image is not None:
     components.html("""<hr style="height:2px;border:none;color:#333;background-color:#333;" /> """)
 
     # predictions
-    results = st.session_state.model.predict(tiles) # predict
+    if 'results' not in st.session_state:
+        st.sessionstate.results = st.session_state.model.predict(tiles) # predict
     
     
     _, row4_col2, row4_col3, _ = st.columns([1,4,4,1], gap="medium")
@@ -154,10 +156,10 @@ if st.session_state.image is not None:
         with row4b_col3:
             if st.button("Next"):
                 if st.session_state.image_counter < len(results):
-                    st.session_state.image_counter -= 1
+                    st.session_state.image_counter += 1
                 
         # Plot predicted image
-        im_bgr = results[st.session_state.image_counter].plot()  # BGR-order numpy array
+        im_bgr = st.sessionstate.results[st.session_state.image_counter].plot()  # BGR-order numpy array
         im_rgb = Image.fromarray(im_bgr[..., ::-1])  # RGB-order PIL image
         st.image(im_rgb)