Update app.py

app.py

@@ -73,16 +73,13 @@ def main():
             loading_dataset_text.text("Loading Dataset...")
             loading_dataset_bar = st.progress(0)
             
-            # Memory profiling
-            mem_usage = memory_usage((load_dataset_with_limit, (dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects), {"limit": dataset_limit}))
-            st.write(f"Memory used for loading the dataset: {mem_usage[-1]:.2f} MB")
             
             # Simulate dataset loading progress
             for i in range(0, 100, 25):
                 time.sleep(0.2)  # Simulate work being done
                 loading_dataset_bar.progress(i + 25)
 
-            # Load dataset and monitor CPU and memory
+            # Load dataset
             df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=dataset_limit)
             
             # Store loaded dataset in session state