Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -28,7 +28,7 @@ class StateManager:
         # Use the second column (col2) for the checkbox
 
         
-        #with col1:
+        with col1:
         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')
         default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
@@ -36,7 +36,7 @@ class StateManager:
 
         # Conditional display of model settings
         with col2:
-            show_model_settings = st.sidebar.checkbox("Show Model Settings", False)
+            show_model_settings = st.checkbox("Show Model Settings", False)
             if show_model_settings:
                 self.display_model_settings()