recompute embedding if missing or state change

app.py

@@ -20,6 +20,11 @@ def process_box(predictor_model, show_mask, radius_width):
     scale           = container_width/width
     scaled_hw = (container_width, int(height * scale))
 
+    if not predictor_model.set_image:
+        np_image = np.asanyarray(bg_image)
+        with st.spinner(text="Extracing embeddings.."):
+            predictor_model.set_image(np_image)
+
     if 'result_image' not in st.session_state:
         st.session_state.result_image = bg_image.resize(scaled_hw)
 
@@ -94,6 +99,11 @@ def process_click(predictor_model, show_mask, radius_width):
     scale           = container_width/width
     scaled_hw = (container_width, int(height * scale))
 
+    if not predictor_model.set_image:
+        np_image = np.asanyarray(bg_image)
+        with st.spinner(text="Extracing embeddings.."):
+            predictor_model.set_image(np_image)
+
     if 'result_image' not in st.session_state:
         st.session_state.result_image = bg_image.resize(scaled_hw)
 
@@ -218,6 +228,10 @@ def image_preprocess_callback(predictor_model, option):
             np_image = np.asanyarray(image)
             with st.spinner(text="Extracing embeddings.."):
                 predictor_model.set_image(np_image)
+        else:
+            if predictor_model:
+                predictor_model.reset_image()
+
         st.session_state.image = image
     else:
         with st.spinner(text="Cleaning up!"):