Update my_model/state_manager.py

my_model/state_manager.py

@@ -22,12 +22,9 @@ class StateManager:
 
 
     def set_up_widgets(self):
+        
         # 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')
@@ -68,7 +65,7 @@ class StateManager:
             free_gpu_resources()
             st.session_state['kbvqa'] = prepare_kbvqa_model()
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
-            #self.update_model_settings(detection_model, confidence_level)
+            
             # Update the previous state with current session state values
             st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}
             
@@ -104,7 +101,7 @@ class StateManager:
         except Exception as e:
             st.error(f"Error reloading detection model: {e}")
 
-    # New methods to be added
+    
     def process_new_image(self, image_key, image, kbvqa):
         if image_key not in st.session_state['images_data']:
             st.session_state['images_data'][image_key] = {