fixed onehotencoder issue1.0

app.py

@@ -26,8 +26,7 @@ def main():
 	<div style="background-color:pink;"><p style="color:white;font-size:75px;padding:10px">Room Occupancy Predictor 💡</p></div>
 	"""
 	st.markdown(html_temp,unsafe_allow_html=True)
-	if st.button("Thanks"):
-		st.balloons()
+
 
 	def file_selector(folder_path='./dataset'):
 		filenames = os.listdir(folder_path)
@@ -156,6 +155,10 @@ def main():
 		imp = SimpleImputer(missing_values = np.nan, strategy="mean")
 		df.loc[:,df.dtypes != 'object'] = imp.fit_transform(df.loc[:,df.dtypes != 'object'])
 
+
+	if st.button("Thanks"):
+		st.balloons()
+
 # One hot encoding for categorical variables
 
 	features = st.multiselect('select features and target variable',df.columns.tolist())
@@ -166,7 +169,7 @@ def main():
 		df.loc[:,x] = le.fit_transform(df[x])
 		onehotencoder = OneHotEncoder()
 		df.loc[:,~cats].join(pd.DataFrame(data=onehotencoder.
-		fit_transform (df.loc[:,cats]).toarray(), columns=onehotencoder.get_feature_names()))
+		fit_transform (df.loc[:,cats]).toarray(), columns=onehotencoder.get_feature_names_out()))
 
 	chosen_target = st.sidebar.selectbox("Please choose target column", (df.columns))