Update my_model/state_manager.py
Browse files
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 |
|