Update app.py

app.py

@@ -21,7 +21,7 @@ os.environ["PYOPENGL_PLATFORM"] = "egl"
 # 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)
+nb_samples = 500
 
 field_names_train = ["sig11", "sig22", "sig12", "U1", "U2", "q"]
 
@@ -92,7 +92,7 @@ def sample_info(sample_id_str, fieldn):
 if __name__ == "__main__":
 
     with gr.Blocks() as demo:
-        d1 = gr.Slider(0, nb_samples, value=0, label="Training sample id", info="Choose between 0 and "+str(nb_samples))
+        d1 = gr.Slider(0, nb_samples-1, value=0, label="Training sample id", info="Choose between 0 and "+str(nb_samples-1))
         d2 = gr.Dropdown(field_names_train, value=field_names_train[0], label="Field name")