add download weights

app.py

@@ -4,11 +4,12 @@ import os
 import torch
 import trimesh
 import sys
-sys.path.append("/home/user/app/cube")
+sys.path.append("cube")
 from cube3d.inference.engine import EngineFast
 from pathlib import Path
 import uuid
 import shutil
+from huggingface_hub import snapshot_download
 
 
 GLOBAL_STATE = {}
@@ -104,10 +105,17 @@ if __name__=="__main__":
     )
 
     args = parser.parse_args()
+    snapshot_download(
+        repo_id="Roblox/cube3d-v0.1",
+        local_dir="./model_weights"
+    )
+    config_path = "./model_weights/shape_tokenizer.safetensors"
+    gpt_ckpt_path = "./model_weights/shape_gpt.safetensors"
+    shape_ckpt_path = "./model_weights/shape_tokenizer.safetensors"
     engine_fast = EngineFast(
-        args.config_path, 
-        args.gpt_ckpt_path, 
-        args.shape_ckpt_path,
+        config_path,
+        gpt_ckpt_path, 
+        shape_ckpt_path,
         device=torch.device("cuda"),
     )
     GLOBAL_STATE["engine_fast"] = engine_fast