Update app.py

app.py

@@ -18,15 +18,15 @@ 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/Rotor37", split="all_samples")
+hf_dataset = load_dataset("PLAID-datasets/VKI-LS59", split="all_samples")
 
-nb_samples = 1000
+nb_samples = 839
 
-field_names_train = ["Density", "Pressure", "Temperature"]
+field_names = ["sdf", "ro", "rou", "rov", "roe", "nut", "mach"]
 
 
 _HEADER_ = '''
-<h2><b>Visualization demo of <a href='https://huggingface.co/datasets/PLAID-datasets/Rotor37' target='_blank'><b>Rotor37 dataset</b></b></h2>
+<h2><b>Visualization demo of <a href='https://huggingface.co/datasets/PLAID-datasets/VKI-LS59' target='_blank'><b>VKI-LS59 dataset</b></b></h2>
 '''
 
 def round_num(num)->str:
@@ -38,8 +38,8 @@ def sample_info(sample_id_str, fieldn):
     plaid_sample = Sample.model_validate(pickle.loads(sample_))
     # plaid_sample = Sample.load_from_dir(f"Tensile2d/dataset/samples/sample_"+str(sample_id_str).zfill(9))
 
-    nodes = plaid_sample.get_nodes()
-    field = plaid_sample.get_field(fieldn)
+    nodes = plaid_sample.get_nodes(base_name="Base_2_2")
+    field = plaid_sample.get_field(fieldn, base_name="Base_2_2")
     # if nodes.shape[1] == 2:
     #     nodes__ = np.zeros((nodes.shape[0],nodes.shape[1]+1))
     #     nodes__[:,:-1] = nodes
@@ -51,7 +51,7 @@ def sample_info(sample_id_str, fieldn):
     
     nb_nodes = nodes.shape[0]
 
-    quads = plaid_sample.get_elements()['QUAD_4']
+    quads = plaid_sample.get_elements(base_name="Base_2_2")['QUAD_4']
     nb_quads = quads.shape[0]
 
     assert field.shape[0] == nb_nodes