Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -22,15 +22,23 @@ 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')
-        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, value=default_confidence, step=0.1, slider_key_name='confidence_level')
-        show_model_settings = st.sidebar.checkbox("Show Model Settings", False)
+        # Create two columns with different widths
+        col1, col2 = st.columns([0.8, 0.2])  # Adjust the ratio as needed
+
+        # Use the second column (col2) for the checkbox
+
+        
+        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
+            self.set_slider_value(text="Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1, slider_key_name='confidence_level')
 
         # Conditional display of model settings
-        if show_model_settings:
-            seld.display_model_settings()
+        with col2:
+            show_model_settings = st.sidebar.checkbox("Show Model Settings", False)
+            if show_model_settings:
+                self.display_model_settings()
 
         
     def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name):