Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -25,8 +25,10 @@ class StateManager:
     
     def set_up_widgets(self):
         st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
-        self.detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
-        default_confidence = 0.2 if self.detection_model == "yolov5" else 0.4
+        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
         self.set_slider_value(text="Select minimum detection confidence level", 
                                 min_value=0.1, 
                                 max_value=0.9,