m7mdal7aj commited on
Commit
d9364fd
1 Parent(s): e00ca5f

Update my_model/utilities/state_manager.py

Browse files
my_model/utilities/state_manager.py CHANGED
@@ -55,8 +55,8 @@ class StateManager:
55
  try:
56
  free_gpu_resources()
57
  st.text("Loading the model, this should take no more than a few minutes, please wait...")
58
- st.session_state['kbvqa'] = prepare_kbvqa_model(st.state_session.detection_model)
59
- st.session_state['kbvqa'].detection_confidence = st.state_session.confidence_level
60
  #self.update_model_settings(detection_model, confidence_level)
61
  st.text("Model is ready for inference.")
62
  free_gpu_resources()
 
55
  try:
56
  free_gpu_resources()
57
  st.text("Loading the model, this should take no more than a few minutes, please wait...")
58
+ st.session_state['kbvqa'] = prepare_kbvqa_model(st.session_state.detection_model)
59
+ st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
60
  #self.update_model_settings(detection_model, confidence_level)
61
  st.text("Model is ready for inference.")
62
  free_gpu_resources()