Upload app_hg.py with huggingface_hub
Browse files
app_hg.py
CHANGED
@@ -51,6 +51,7 @@ parser.add_argument("--save_memory", default=False) # , action="store_true")
|
|
51 |
parser.add_argument("--device", default="cuda:0", type=str)
|
52 |
args = parser.parse_args()
|
53 |
|
|
|
54 |
def find_cuda():
|
55 |
# Check if CUDA_HOME or CUDA_PATH environment variables are set
|
56 |
cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
|
@@ -114,9 +115,8 @@ CONST_MAX_QUEUE = 1
|
|
114 |
CONST_SERVER = '0.0.0.0'
|
115 |
|
116 |
CONST_HEADER = '''
|
117 |
-
<h2><b>Official 🤗 Gradio Demo</b></h2><h2><a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'><b>Hunyuan3D-1.0: A Unified Framework for Text-to-3D and Image-to-3D
|
118 |
-
|
119 |
-
Code: <a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'>GitHub</a>. Techenical report: <a href='https://arxiv.org/abs/placeholder' target='_blank'>ArXiv</a>.
|
120 |
|
121 |
❗️❗️❗️**Important Notes:**
|
122 |
- By default, our demo can export a .obj mesh with vertex colors or a .glb mesh.
|
@@ -126,7 +126,7 @@ Code: <a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'>GitHub</a
|
|
126 |
'''
|
127 |
|
128 |
CONST_CITATION = r"""
|
129 |
-
If
|
130 |
---
|
131 |
📝 **Citation**
|
132 |
If you find our work useful for your research or applications, please cite using this bibtex:
|
|
|
51 |
parser.add_argument("--device", default="cuda:0", type=str)
|
52 |
args = parser.parse_args()
|
53 |
|
54 |
+
@spaces.GPU
|
55 |
def find_cuda():
|
56 |
# Check if CUDA_HOME or CUDA_PATH environment variables are set
|
57 |
cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
|
|
|
115 |
CONST_SERVER = '0.0.0.0'
|
116 |
|
117 |
CONST_HEADER = '''
|
118 |
+
<h2><b>Official 🤗 Gradio Demo</b></h2><h2><a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'><b>Tencent Hunyuan3D-1.0: A Unified Framework for Text-to-3D and Image-to-3D Generation</b></a></h2>
|
119 |
+
Code: <a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'>GitHub</a>. Techenical report: <a href='https://arxiv.org/pdf/2411.02293' target='_blank'>ArXiv</a>.
|
|
|
120 |
|
121 |
❗️❗️❗️**Important Notes:**
|
122 |
- By default, our demo can export a .obj mesh with vertex colors or a .glb mesh.
|
|
|
126 |
'''
|
127 |
|
128 |
CONST_CITATION = r"""
|
129 |
+
If Hunyuan3D-1.0 is helpful, please help to ⭐ the <a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'>Github Repo</a>. Thanks! [![GitHub Stars](https://img.shields.io/github/stars/tencent/Hunyuan3D-1?style=social)](https://github.com/tencent/Hunyuan3D-1)
|
130 |
---
|
131 |
📝 **Citation**
|
132 |
If you find our work useful for your research or applications, please cite using this bibtex:
|