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 |
|