test-final

app.py

@@ -526,6 +526,7 @@ class DynamicsVisualizer:
         self.friction = friction
 
     def reload_model(self, num_steps):  # only change num_steps
+        from pgnd.sim import CacheDiffSimWithFrictionBatch
         self.cfg.sim.num_steps = num_steps
         sim = CacheDiffSimWithFrictionBatch(self.cfg, num_steps, 1, self.wp_device, requires_grad=True)
         self.sim = sim
@@ -1162,13 +1163,12 @@ class DynamicsVisualizer:
 
     def launch(self, share=False):
         
-        preprocess_metadata = gr.State(self.preprocess_metadata)
-        state = gr.State(self.state)
-        params = gr.State(self.params)
-        table_params = gr.State(self.table_params)
-        gripper_params = gr.State(self.gripper_params)
-
         with gr.Blocks() as app:
+            preprocess_metadata = gr.State(self.preprocess_metadata)
+            state = gr.State(self.state)
+            params = gr.State(self.params)
+            table_params = gr.State(self.table_params)
+            gripper_params = gr.State(self.gripper_params)
 
             with gr.Row():
                 gr.Markdown("# Particle-Grid Neural Dynamics for Learning Deformable Object Models from RGB-D Videos")

src/experiments/log/gs/temp/gs_pred.splat

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a881c1547ec6042bf70b5b2270683bcdbc27151042736b2eca5198a5e4cb4ff3
+oid sha256:9e66dd4cc45e6a67802892bf1173190406d3403ecd2416313fbc5b94eff0898a
 size 7675392