Update app.py

app.py

@@ -25,7 +25,6 @@ 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 = "VAST-AI/DetailGen3D" # 似乎还没有
 
 
 MARKDOWN = """
@@ -77,7 +76,7 @@ EXAMPLES = [
 os.makedirs(TMP_DIR, exist_ok=True)
 
 local_dir = "pretrained_weights/DetailGen3D"
-snapshot_download(repo_id=REPO_ID, local_dir=local_dir)
+snapshot_download(repo_id="VAST-AI/DetailGen3D", local_dir=local_dir)
 pipeline = DetailGen3DPipeline.from_pretrained(
     local_dir
 ).to(DEVICE, dtype=DTYPE)