Upload folder using huggingface_hub

README.md

@@ -1,3 +1,15 @@
+---
+title: Hunyuan3D-2.0
+emoji: 😻
+colorFrom: purple
+colorTo: red
+sdk: gradio
+sdk_version: 3.39.0
+app_file: app_hg.py
+pinned: false
+short_description: Text-to-3D and Image-to-3D Generation
+---
+
 [中文阅读](README_zh_cn.md)
 
 <p align="center">

gradio_app.py

@@ -7,6 +7,8 @@ from glob import glob
 import gradio as gr
 import torch
 
+import spaces
+
 
 def get_example_img_list():
     print('Loading example img list ...')
@@ -77,7 +79,7 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         </div>
     """
 
-
+@spaces.GPU()
 def _gen_shape(
     caption,
     image,
@@ -134,7 +136,7 @@ def _gen_shape(
     stats['time'] = time_meta
     return mesh, save_folder
 
-
+@spaces.GPU()
 def generation_all(
     caption,
     image,
@@ -167,7 +169,7 @@ def generation_all(
         model_viewer_html_textured,
     )
 
-
+@spaces.GPU()
 def shape_generation(
     caption,
     image,
@@ -351,8 +353,4 @@ if __name__ == '__main__':
     face_reduce_worker = FaceReducer()
 
     demo = build_app()
-    demo.queue().launch(
-        server_name='0.0.0.0',
-        server_port=args.port,
-        allowed_paths=[SAVE_DIR],
-    )
+    demo.queue().launch()

hy3dgen/shapegen/pipelines.py

@@ -215,8 +215,7 @@ class Hunyuan3DDiTPipeline:
                 try:
                     import huggingface_hub
                     path = huggingface_hub.snapshot_download(
-                        repo_id="tencent/Hunyuan3D-2", 
-                        local_dir=base_dir, 
+                        repo_id=original_model_path, 
                         resume_download=True
                     )
                     model_path = os.path.join(path, 'hunyuan3d-dit-v2-0')

hy3dgen/texgen/pipelines.py

@@ -71,8 +71,7 @@ class Hunyuan3DPaintPipeline:
                 try:
                     import huggingface_hub
                     path = huggingface_hub.snapshot_download(
-                        repo_id="tencent/Hunyuan3D-2", 
-                        local_dir=base_dir, 
+                        repo_id=original_model_path, 
                         resume_download=True
                     )
                     delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')