m7mdal7aj commited on
Commit
6d688ae
1 Parent(s): aac0218

Update my_model/state_manager.py

Browse files
Files changed (1) hide show
  1. my_model/state_manager.py +3 -3
my_model/state_manager.py CHANGED
@@ -220,7 +220,7 @@ class StateManager:
220
  try:
221
  free_gpu_resources()
222
  if self.is_model_loaded:
223
- st.text("tttttt")
224
  st.write(st.session_state['detection_model'] == st.session_state['previous_state']['detection_model'])
225
  st.write(st.session_state['method'] == st.session_state['previous_state']['method'])
226
  st.write(st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level'])
@@ -229,9 +229,9 @@ class StateManager:
229
  if st.session_state['detection_model'] == st.session_state['previous_state']['detection_model']\
230
  and st.session_state['method'] == st.session_state['previous_state']['method'] \
231
  and st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level']:
232
- st.text("aaaaa")
233
  st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
234
- st.text("HHHHHH")
235
  return # only update the confidence level
236
 
237
  prepare_kbvqa_model(only_reload_detection_model=True)
 
220
  try:
221
  free_gpu_resources()
222
  if self.is_model_loaded:
223
+
224
  st.write(st.session_state['detection_model'] == st.session_state['previous_state']['detection_model'])
225
  st.write(st.session_state['method'] == st.session_state['previous_state']['method'])
226
  st.write(st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level'])
 
229
  if st.session_state['detection_model'] == st.session_state['previous_state']['detection_model']\
230
  and st.session_state['method'] == st.session_state['previous_state']['method'] \
231
  and st.session_state['confidence_level'] != st.session_state['previous_state']['confidence_level']:
232
+
233
  st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
234
+ st.write("reload func", st.session_state['kbvqa'].detection_confidence )
235
  return # only update the confidence level
236
 
237
  prepare_kbvqa_model(only_reload_detection_model=True)