Update app.py

app.py

@@ -82,13 +82,11 @@ EXAMPLES = [
 
 os.makedirs(TMP_DIR, exist_ok=True)
 
-dtype = torch.float16
-
 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 +128,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