Update my_model/state_manager.py

my_model/state_manager.py

@@ -19,11 +19,11 @@ class StateManager:
         if "previous_state" not in st.session_state:
             st.session_state['previous_state'] = {'method': None, 'detection_model': None, 'confidence_level': None}
         if 'method' not in st.session_state:
-            st.session_state['method'] = None
-        if 'method' not in st.session_state:
-            st.session_state['confidence_level'] = None
-        if 'method' not in st.session_state:
-            st.session_state['detection_model'] = None
+            st.session_state['method'] = "Fine-Tuned Model"
+        if 'confidence_level' not in st.session_state:
+            st.session_state['confidence_level'] = 0.35
+        if 'detection_model' not in st.session_state:
+            st.session_state['detection_model'] = 'detic'
         if 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
         if 'kbvqa' not in st.session_state: