Update pages/2_🚀_QuickML.py

pages/2_🚀_QuickML.py

@@ -64,23 +64,22 @@ st.markdown("QuickML is a tool that helps you to build a Machine Learning model
 st.divider()
 
 
-st.header("Upload Your CSV File", anchor=False)
+st.header("Upload Your CSV File")
 uploaded_file = st.file_uploader("Upload the Dataset", type=["csv", "xlsx", "xls"])
 st.divider()
 
 if uploaded_file:
-    # Read the CSV File
     if st.session_state.df is None:
-         # Read CSV or Excel file
+        # Read CSV or Excel file
         if uploaded_file.name.endswith('.csv'):
-            return pd.read_csv(uploaded_file)
-        elif uploaded_file.name.endswith('.xlsx') or file.name.endswith('.xls'):
-            return pd.read_excel(uploaded_file)
+            st.session_state.df = pd.read_csv(uploaded_file)
+        elif uploaded_file.name.endswith('.xlsx') or uploaded_file.name.endswith('.xls'):
+            st.session_state.df = pd.read_excel(uploaded_file)
         else:
-            raise ValueError("Unsupported file format. Only CSV and Excel files are supported.")
+            st.error("Unsupported file format. Only CSV and Excel files are supported.")
     else:
-        df = st.session_state.df 
-        # st.dataframe(df)
+        df = st.session_state.df
+        st.dataframe(df)
         new_line()