fix to add proper split

app.py

@@ -50,7 +50,7 @@ for subset in subsets:
             "total_magnetization",
         ],
     )
-    datasets.append(dataset)
+    datasets.append(dataset['train'])
 
 display_columns = [
     "chemical_formula_descriptive",