m7mdal7aj commited on
Commit
87db14a
1 Parent(s): 094a401

Update my_model/state_manager.py

Browse files
Files changed (1) hide show
  1. my_model/state_manager.py +2 -3
my_model/state_manager.py CHANGED
@@ -202,7 +202,7 @@ class StateManager:
202
  if key == 'confidence_level':
203
  continue # confidence_level tracker is separate
204
  if key in st.session_state and st.session_state[key] != st.session_state['previous_state'][key]:
205
- st.write("change check", 'now', st.session_state[key], "prev", st.session_state['previous_state'][key] )
206
  return True # Found a change
207
  else: return False # No changes found
208
 
@@ -298,8 +298,7 @@ class StateManager:
298
  st.text("Analyzing the image .. please wait")
299
  caption = st.session_state['kbvqa'].get_caption(img)
300
  image_with_boxes, detected_objects_str = st.session_state['kbvqa'].detect_objects(img)
301
- st.write("inside analyze", st.session_state['confidence_level'])
302
- st.write(detected_objects_str)
303
  return caption, detected_objects_str, image_with_boxes
304
 
305
 
 
202
  if key == 'confidence_level':
203
  continue # confidence_level tracker is separate
204
  if key in st.session_state and st.session_state[key] != st.session_state['previous_state'][key]:
205
+
206
  return True # Found a change
207
  else: return False # No changes found
208
 
 
298
  st.text("Analyzing the image .. please wait")
299
  caption = st.session_state['kbvqa'].get_caption(img)
300
  image_with_boxes, detected_objects_str = st.session_state['kbvqa'].detect_objects(img)
301
+ free_gpu_resources()
 
302
  return caption, detected_objects_str, image_with_boxes
303
 
304