444

app.py

@@ -20,7 +20,7 @@ from datasets import load_dataset
 @st.cache_data
 def load_data():
     train_df = load_dataset("kheejay88/phone_price_classification_train")["train"].to_pandas()
-    test_df = load_dataset("kheejay88/phone_price_classification_test")["train"].to_pandas()
+    test_df = load_dataset("kheejay88/phone_price_classification_test")["test"].to_pandas()
     return train_df, test_df
 
 train_df, test_df = load_data()