Update app.py

app.py

@@ -88,7 +88,7 @@ local_dir = "pretrained_weights/DetailGen3D"
 snapshot_download(repo_id=REPO_ID, local_dir=local_dir)
 pipeline = DetailGen3DPipeline.from_pretrained(
     local_dir
-).to(device, dtype=dtype)
+).to(DEVICE, dtype=dtype)
 
 
 def load_mesh(mesh_path, num_pc=20480):
@@ -130,7 +130,7 @@ def run_detailgen3d(
     sampled_points, grid_size, bbox_size = generate_dense_grid_points(
         bbox_min=box_min, bbox_max=box_max, octree_depth=8, indexing="ij"
     )
-    sampled_points = torch.FloatTensor(sampled_points).to(device, dtype=dtype)
+    sampled_points = torch.FloatTensor(sampled_points).to(DEVICE, dtype=dtype)
     sampled_points = sampled_points.unsqueeze(0).repeat(batch_size, 1, 1)
 
     # inference pipeline