Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
-# import pickle
-# from datasets import load_from_disk
+import pickle
+from datasets import load_dataset
 from plaid.containers.sample import Sample
-# import pyvista as pv
+
 
 import numpy as np
 import pyrender
@@ -15,24 +15,23 @@ import os
 os.environ["PYOPENGL_PLATFORM"] = "egl"
 
 
-os.system("wget https://zenodo.org/records/10124594/files/Tensile2d.tar.gz")
-os.system("tar -xvf Tensile2d.tar.gz")
+# os.system("wget https://zenodo.org/records/10124594/files/Tensile2d.tar.gz")
+# os.system("tar -xvf Tensile2d.tar.gz")
 
-# FOLDER = "plot"
+hf_dataset = load_dataset(“PLAID-datasets/Tensile2d”, split=“all_samples”)
 
-# dataset = load_from_disk("Rotor37")
 
 field_names_train = ["sig11", "sig22", "sig12", "U1", "U2", "q"]
 
 
 
 
-
-
-
 def sample_info(sample_id_str, fieldn):
 
-    plaid_sample = Sample.load_from_dir(f"Tensile2d/dataset/samples/sample_"+str(sample_id_str).zfill(9))
+    sample_ = hf_dataset[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))
+
     nodes = plaid_sample.get_nodes()
     field = plaid_sample.get_field(fieldn)
     if nodes.shape[1] == 2: