Update app.py

app.py

@@ -15,11 +15,12 @@ from sklearn.metrics import accuracy_score
 from imblearn.over_sampling import SMOTE
 import joblib
 import time
+from datasets import load_dataset
 
 # Load dataset
 @st.cache_data
 def load_data():
-    df = pd.read_csv("kheejay88/water_potability")
+    df = load_dataset("kheejay88/water_potability")["train"].to_pandas()
     return df
 
 df = load_data()