delete prediction if we change image

app.py

@@ -26,7 +26,7 @@ def main():
     if cropped_image is not None:
         get_score_threshold(is_mobile)
         if st.button("🚀 Launch Prediction"):
-            launch_prediction(cropped_image, st.session_state.score_threshold, is_mobile, screen_width)
+            st.session_state.image = launch_prediction(cropped_image, st.session_state.score_threshold, is_mobile, screen_width)
             st.session_state.original_prediction = st.session_state.prediction.copy()
             st.rerun()
     
@@ -34,6 +34,12 @@ def main():
         #if st.button("🔄 Refresh image"):
             #st.rerun()
 
+        if st.session_state.image != cropped_image:
+            print('Image has changed')
+            #delete the prediction
+            del st.session_state.prediction
+            return
+
         with st.expander("Show result of prediction"):
             with st.spinner('Waiting for result display...'):
                 display_options(st.session_state.crop_image, st.session_state.score_threshold, is_mobile, int(5/6 * screen_width))

modules/streamlit_utils.py

@@ -293,11 +293,12 @@ def get_score_threshold(is_mobile):
 def launch_prediction(cropped_image, score_threshold, is_mobile, screen_width):
     st.session_state.crop_image = cropped_image
     with st.spinner('Processing...'):
-        perform_inference(
+        image, _ , _ = perform_inference(
             st.session_state.model_object, st.session_state.model_arrow, st.session_state.crop_image,
             score_threshold, is_mobile, screen_width, iou_threshold=0.3, distance_treshold=30, percentage_text_dist_thresh=0.5
         )
         st.balloons()    
+        return image
     
 
 def modify_results(percentage_text_dist_thresh=0.5):