Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -14,8 +14,8 @@ class StateManager:
             st.session_state['images_data'] = {}
         if 'method' not in st.session_state:
             st.session_state['method'] = None
-        if 'detection_model' not in st.session_state:
-            st.session_state['detection_model'] = None
+       # if 'detection_model' not in st.session_state:
+        #    st.session_state['detection_model'] = None
         if 'kbvqa' not in st.session_state:
             st.session_state['kbvqa'] = None
         if 'confidence_level' not in st.session_state:
@@ -26,9 +26,9 @@ class StateManager:
     def set_up_widgets(self):
         st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
         detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
-        st.write("Detection Model")
-        st.write(detection_model)
-        default_confidence = 0.2 if detection_model == "yolov5" else 0.4
+        st.write('detection model')
+        st.write(st.session_state.detection_model)
+        default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
         self.set_slider_value(text="Select minimum detection confidence level", 
                                 min_value=0.1, 
                                 max_value=0.9, 
@@ -36,7 +36,8 @@ class StateManager:
                                 step=0.1, 
                                 slider_key_name='confidence_level'
                                                     )
-        
+        st.write("Detection Confidence")
+        st.write(st.session_state.confidence_level)
 
     def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name):