m7mdal7aj commited on
Commit
d6a4897
1 Parent(s): 09ff02d

Update my_model/utilities/state_manager.py

Browse files
my_model/utilities/state_manager.py CHANGED
@@ -36,7 +36,7 @@ class StateManager:
36
  def display_model_settings(self):
37
  st.write("#### Current Model Settings:")
38
  data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa']]
39
- df = pd.DataFrame(data, index=False)
40
  styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'white', 'border-color': 'white'})
41
  st.table(styled_df)
42
 
 
36
  def display_model_settings(self):
37
  st.write("#### Current Model Settings:")
38
  data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa']]
39
+ df = pd.DataFrame(data)
40
  styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'white', 'border-color': 'white'})
41
  st.table(styled_df)
42