fix bug

app.py

@@ -120,6 +120,7 @@ if  st.session_state.option is not None and \
     st.session_state.data_item = process_ct_gt(st.session_state.option)
     st.session_state.reset_demo_case = False
     st.session_state.preds_3D = None
+    st.session_state.preds_3D_ori = None
 
 prompt_col1, prompt_col2 = st.columns(2)
 
@@ -288,6 +289,7 @@ with col1:
                  disabled=(st.session_state.option is None or (len(st.session_state.points)==0 and not st.session_state.use_box_prompt and st.session_state.preds_3D is None))):
         clear_prompts()
         st.session_state.preds_3D = None
+        st.session_state.preds_3D_ori = None
         st.rerun()
 
 with col2: