Update pages/2_🚀_QuickML.py

pages/2_🚀_QuickML.py

@@ -72,12 +72,26 @@ if uploaded_file:
 
     # Read the CSV File
     if st.session_state.df is None:
-        df = pd.read_csv(uploaded_file)
+         # Read CSV or Excel file
+        if file.name.endswith('.csv'):
+            return pd.read_csv(file)
+        elif file.name.endswith('.xlsx') or file.name.endswith('.xls'):
+            return pd.read_excel(file)
+        else:
+            raise ValueError("Unsupported file format. Only CSV and Excel files are supported.")
     else:
         df = st.session_state.df 
         # st.dataframe(df)
         new_line()
 
+     # Read CSV or Excel file
+    if file.name.endswith('.csv'):
+        return pd.read_csv(file)
+    elif file.name.endswith('.xlsx') or file.name.endswith('.xls'):
+        return pd.read_excel(file)
+    else:
+        raise ValueError("Unsupported file format. Only CSV and Excel files are supported.")
+
     # The Dataset
     st.subheader("🎬 Dataset", anchor=False)
     new_line()