Update app.py

app.py

@@ -18,13 +18,11 @@ os.environ["PYOPENGL_PLATFORM"] = "egl"
 # os.system("wget https://zenodo.org/records/10124594/files/Tensile2d.tar.gz")
 # os.system("tar -xvf Tensile2d.tar.gz")
 
-hf_dataset = load_dataset("PLAID-datasets/Tensile2d", split="all_samples")
-
-nb_samples = 500
-
-field_names_train = ["sig11", "sig22", "sig12", "U1", "U2", "q"]
+hf_dataset = load_dataset("PLAID-datasets/Rotor37", split="all_samples")
 
+nb_samples = 1000
 
+field_names_train = ["Density", "Pressure", "Temperature"]
 
 
 def sample_info(sample_id_str, fieldn):