Update app.py

app.py

@@ -7,12 +7,12 @@ from sklearn.metrics import mean_squared_error, r2_score
 import altair as alt
 import time
 import zipfile
+import os
 
 # Page title
 st.set_page_config(page_title='ML Model Building', page_icon='🤖', layout='wide')  # Set layout to wide for better use of space
 st.title('🤖 ML Model Building')
 
-
 with st.expander('About this app'):
   st.markdown('**What can this app do?**')
   st.info('This app allow users to build a machine learning (ML) model in an end-to-end workflow. Particularly, this encompasses data upload, data pre-processing, ML model building and post-model analysis.')
@@ -32,7 +32,6 @@ with st.expander('About this app'):
 - Streamlit for user interface
   ''', language='markdown')
 
-
 # Sidebar for accepting input parameters
 with st.sidebar:
     # Load data
@@ -130,14 +129,8 @@ if uploaded_file or example_data:
         st.write("Displaying performance metrics ...")
         time.sleep(sleep_time)
         parameter_criterion_string = ' '.join([x.capitalize() for x in parameter_criterion.split('_')])
-        #if 'Mse' in parameter_criterion_string:
-        #    parameter_criterion_string = parameter_criterion_string.replace('Mse', 'MSE')
         rf_results = pd.DataFrame(['Random forest', train_mse, train_r2, test_mse, test_r2]).transpose()
         rf_results.columns = ['Method', f'Training {parameter_criterion_string}', 'Training R2', f'Test {parameter_criterion_string}', 'Test R2']
-        # Convert objects to numerics
-        for col in rf_results.columns:
-            rf_results[col] = pd.to_numeric(rf_results[col], errors='ignore')
-        # Round to 3 digits
         rf_results = rf_results.round(3)
         
     status.update(label="Status", state="complete", expanded=False)
@@ -244,7 +237,6 @@ if uploaded_file or example_data:
                   )
         st.altair_chart(scatter, theme='streamlit', use_container_width=True)
 
-    
-# Ask for CSV upload if none is detected
 else:
     st.warning('👈 Upload a CSV file or click *"Load example data"* to get started!')
+