Update my_model/utilities/state_manager.py
Browse files
my_model/utilities/state_manager.py
CHANGED
@@ -73,7 +73,7 @@ class StateManager:
|
|
73 |
def is_model_loaded(self):
|
74 |
return 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None
|
75 |
|
76 |
-
def reload_detection_model(self
|
77 |
try:
|
78 |
free_gpu_resources()
|
79 |
if self.is_model_loaded():
|
|
|
73 |
def is_model_loaded(self):
|
74 |
return 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None
|
75 |
|
76 |
+
def reload_detection_model(self):
|
77 |
try:
|
78 |
free_gpu_resources()
|
79 |
if self.is_model_loaded():
|