m7mdal7aj commited on
Commit
9da8a28
1 Parent(s): a1f1f0b

Update my_model/state_manager.py

Browse files
Files changed (1) hide show
  1. my_model/state_manager.py +1 -1
my_model/state_manager.py CHANGED
@@ -422,7 +422,7 @@ class StateManager:
422
 
423
  """
424
  self.col3.write("##### Current Model Settings:")
425
- data = [{'Setting': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa', 'previous_state', 'settings_changed', 'loading_in_progress', 'model_loaded', 'time_taken_to_load_model' ]]
426
  df = pd.DataFrame(data).reset_index(drop=True)
427
  return self.col3.write(df)
428
 
 
422
 
423
  """
424
  self.col3.write("##### Current Model Settings:")
425
+ data = [{'Setting': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa', 'previous_state', 'settings_changed', 'loading_in_progress', 'model_loaded', 'time_taken_to_load_model', 'images_data' ]]
426
  df = pd.DataFrame(data).reset_index(drop=True)
427
  return self.col3.write(df)
428