Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -15,7 +15,7 @@ class StateManager:
         if 'method' not in st.session_state:
             st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
         if 'detection_model' not in st.session_state:
-            st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
+            self.detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
         if 'kbvqa' not in st.session_state:
             st.session_state['kbvqa'] = None