KB-VQA-E

Sleeping

m7mdal7aj commited on Feb 29

Commit

867e680

•

1 Parent(s): b0349d7

Update my_model/state_manager.py

Files changed (1) hide show

my_model/state_manager.py CHANGED Viewed

@@ -21,7 +21,7 @@ class StateManager:
         if 'method' not in st.session_state:
             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:

         if 'method' not in st.session_state:
             st.session_state['method'] = "Fine-Tuned Model"
         if 'confidence_level' not in st.session_state:
+            st.session_state['confidence_level'] = 0.4
         if 'detection_model' not in st.session_state:
             st.session_state['detection_model'] = 'detic'
         if 'images_data' not in st.session_state: