Spaces:
Sleeping
Sleeping
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():
|