Update app.py

app.py

@@ -21,7 +21,7 @@ MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp")
 DTYPE = torch.bfloat16
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-REPO_ID = "" # 似乎还没有
+REPO_ID = "VAST-AI/DetailGen3D" # 似乎还没有
 
 MARKDOWN = """
 ## Generating geometry details guided by reference image with [DetailGen3D](https://detailgen3d.github.io/DetailGen3D/)
@@ -85,9 +85,10 @@ os.makedirs(TMP_DIR, exist_ok=True)
 device = "cuda"
 dtype = torch.float16
 
+local_dir = "pretrained_weights/DetailGen3D"
+snapshot_download(repo_id=REPO_ID, local_dir=local_dir)
 pipeline = DetailGen3DPipeline.from_pretrained(
-    "VAST-AI/DetailGen3D",
-    low_cpu_mem_usage=False
+    local_dir
 ).to(device, dtype=dtype)