m7mdal7aj commited on
Commit
cd3678b
1 Parent(s): 6d0a61c

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, detection_model, confidence_level):
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():