Upload app_hg.py with huggingface_hub

app_hg.py

@@ -30,6 +30,7 @@ import torch
 import numpy as np
 from PIL import Image
 from einops import rearrange
+from huggingface_hub import snapshot_download
 
 from infer import seed_everything, save_gif
 from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
@@ -43,10 +44,40 @@ parser.add_argument("--use_lite", default=False, action="store_true")
 parser.add_argument("--mv23d_cfg_path", default="./svrm/configs/svrm.yaml", type=str)
 parser.add_argument("--mv23d_ckt_path", default="weights/svrm/svrm.safetensors", type=str)
 parser.add_argument("--text2image_path", default="weights/hunyuanDiT", type=str)
-parser.add_argument("--save_memory", default=False, action="store_true")
+parser.add_argument("--save_memory", default=True) # , action="store_true")
 parser.add_argument("--device", default="cuda:0", type=str)
 args = parser.parse_args()
 
+def download_models():
+    # Create weights directory if it doesn't exist
+    os.makedirs("weights", exist_ok=True)
+    os.makedirs("weights/hunyuanDiT", exist_ok=True)
+
+    # Download Hunyuan3D-1 model
+    try:
+        snapshot_download(
+            repo_id="tencent/Hunyuan3D-1",
+            local_dir="./weights",
+            resume_download=True
+        )
+        print("Successfully downloaded Hunyuan3D-1 model")
+    except Exception as e:
+        print(f"Error downloading Hunyuan3D-1: {e}")
+
+    # Download HunyuanDiT model
+    try:
+        snapshot_download(
+            repo_id="Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled",
+            local_dir="./weights/hunyuanDiT",
+            resume_download=True
+        )
+        print("Successfully downloaded HunyuanDiT model")
+    except Exception as e:
+        print(f"Error downloading HunyuanDiT: {e}")
+
+# Download models before starting the app
+download_models()
+
 ################################################################
 
 CONST_PORT = 8080