Update app.py

app.py

@@ -18,14 +18,17 @@ 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")
+hf_dataset = load_dataset("fabiencasenave/2D_Multiscale_Hyperelasticity", split="all_samples")
 
-nb_samples = 500
 
-field_names_train = ["sig11", "sig22", "sig12", "U1", "U2", "q"]
+ids_train = hf_dataset.description["split"]["DOE_train"]
+nb_samples = len(ids_train)
+
+
+field_names_train = ["u1", "u2", "P11", "P12", "P22", "P21", "psi"]
 
 _HEADER_ = '''
-<h2><b>Visualization demo of <a href='https://huggingface.co/datasets/PLAID-datasets/Tensile2d' target='_blank'><b>Tensile2d dataset</b></b></h2>
+<h2><b>Visualization demo of <a href='https://huggingface.co/datasets/PLAID-datasets/2D_Multiscale_Hyperelasticity' target='_blank'><b>2D_Multiscale_Hyperelasticity dataset</b></b></h2>
 '''
 
 
@@ -34,7 +37,7 @@ def round_num(num)->str:
 
 def sample_info(sample_id_str, fieldn):
 
-    sample_ = hf_dataset[int(sample_id_str)]["sample"]
+    sample_ = hf_dataset[ids_train[int(sample_id_str)]]["sample"]
     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))