Update my_model/state_manager.py

my_model/state_manager.py

@@ -55,7 +55,7 @@ class StateManager:
  
         show_model_settings = self.col3.checkbox("Show Model Settings", True,  disabled=self.is_widget_disabled)
         if show_model_settings:
-            self.display_model_settings()
+            self.display_model_settings
 
 
         
@@ -96,7 +96,7 @@ class StateManager:
         """
         return self.has_state_changed()
 
-    
+    @property
     def display_model_settings(self):
         """
         Displays a table of current model settings in the third column.
@@ -104,9 +104,9 @@ class StateManager:
         """
         self.col3.write("##### Current Model Settings:")
         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' ]]
-        df = pd.DataFrame(data)
+        df = pd.DataFrame(data).reset_index(drop=True)
         #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')]}])
-        self.col3.write(df)
+        return self.col3.write(df)
 
     
     def display_session_state(self):