Update my_model/state_manager.py
Browse files
my_model/state_manager.py
CHANGED
@@ -100,7 +100,7 @@ class StateManager:
|
|
100 |
Uses formatted HTML to style the table for better readability.
|
101 |
"""
|
102 |
self.col3.write("##### Current Model Settings:")
|
103 |
-
data = [{'
|
104 |
df = pd.DataFrame(data)
|
105 |
styled_df = df.style.set_properties(**{'background-color': 'white', 'color': 'black', 'border-color': 'black'}).set_table_styles([{'selector': 'th','props': [('background-color', 'gray'), ('font-weight', 'bold')]}])
|
106 |
self.col3.write(df)
|
|
|
100 |
Uses formatted HTML to style the table for better readability.
|
101 |
"""
|
102 |
self.col3.write("##### Current Model Settings:")
|
103 |
+
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' ]]
|
104 |
df = pd.DataFrame(data)
|
105 |
styled_df = df.style.set_properties(**{'background-color': 'white', 'color': 'black', 'border-color': 'black'}).set_table_styles([{'selector': 'th','props': [('background-color', 'gray'), ('font-weight', 'bold')]}])
|
106 |
self.col3.write(df)
|