Update my_model/utilities/state_manager.py
Browse files
my_model/utilities/state_manager.py
CHANGED
@@ -77,7 +77,7 @@ class StateManager:
|
|
77 |
try:
|
78 |
free_gpu_resources()
|
79 |
if self.is_model_loaded():
|
80 |
-
prepare_kbvqa_model(
|
81 |
st.session_state['kbvqa'].detection_confidence = confidence_level
|
82 |
#self.update_model_settings(detection_model, confidence_level)
|
83 |
free_gpu_resources()
|
|
|
77 |
try:
|
78 |
free_gpu_resources()
|
79 |
if self.is_model_loaded():
|
80 |
+
prepare_kbvqa_model(only_reload_detection_model=True)
|
81 |
st.session_state['kbvqa'].detection_confidence = confidence_level
|
82 |
#self.update_model_settings(detection_model, confidence_level)
|
83 |
free_gpu_resources()
|