m7mdal7aj commited on
Commit
cc825df
1 Parent(s): 3689b26

Update my_model/utilities/state_manager.py

Browse files
my_model/utilities/state_manager.py CHANGED
@@ -46,7 +46,7 @@ class StateManager:
46
  try:
47
  free_gpu_resources()
48
  st.text("Loading the model, this should take no more than a few minutes, please wait...")
49
- st.session_state['kbvqa'] = prepare_kbvqa_model(st.session_state.detection_model)
50
  st.text("AAAAAAA")
51
  st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
52
  #self.update_model_settings(detection_model, confidence_level)
 
46
  try:
47
  free_gpu_resources()
48
  st.text("Loading the model, this should take no more than a few minutes, please wait...")
49
+ st.session_state['kbvqa'] = prepare_kbvqa_model()
50
  st.text("AAAAAAA")
51
  st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
52
  #self.update_model_settings(detection_model, confidence_level)