Update app.py

app.py

@@ -18,8 +18,10 @@ 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("PLAID-datasets/Tensile2d", split="all_samples")
+hf_dataset = load_dataset("fabiencasenave/Tensile2d_test_2", split="all_samples")
 
+nb_samples = len(hf_dataset)
 
 field_names_train = ["sig11", "sig22", "sig12", "U1", "U2", "q"]
 
@@ -87,7 +89,7 @@ def sample_info(sample_id_str, fieldn):
 if __name__ == "__main__":
 
     with gr.Blocks() as demo:
-        d1 = gr.Slider(0, 499, value=0, label="Training sample id", info="Choose between 0 and 499")
+        d1 = gr.Slider(0, nb_samples, value=0, label="Training sample id", info=f"Choose between 0 and {nb_samples}")
         d2 = gr.Dropdown(field_names_train, value=field_names_train[0], label="Field name")