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(
|
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)
|