Spaces:
Running
on
Zero
Running
on
Zero
fix: Update app
Browse files- README.md +3 -3
- app.py +2 -1
- requirements.txt +6 -5
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
emoji: 🏢
|
4 |
colorFrom: indigo
|
5 |
colorTo: blue
|
@@ -8,9 +8,9 @@ sdk_version: 4.44.1
|
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
license: mit
|
11 |
-
short_description:
|
12 |
---
|
13 |
|
14 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
15 |
|
16 |
-
Paper:
|
|
|
1 |
---
|
2 |
+
title: Hi3DGen
|
3 |
emoji: 🏢
|
4 |
colorFrom: indigo
|
5 |
colorTo: blue
|
|
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
license: mit
|
11 |
+
short_description: High-fidelity 3D Geometry Generation from images
|
12 |
---
|
13 |
|
14 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
15 |
|
16 |
+
Paper:
|
app.py
CHANGED
@@ -223,9 +223,10 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
|
|
223 |
if __name__ == "__main__":
|
224 |
# Initialize pipeline
|
225 |
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
|
|
226 |
|
227 |
# Initialize normal predictor
|
228 |
-
|
229 |
|
230 |
# Launch the app
|
231 |
demo.launch()
|
|
|
223 |
if __name__ == "__main__":
|
224 |
# Initialize pipeline
|
225 |
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
226 |
+
pipeline.cuda()
|
227 |
|
228 |
# Initialize normal predictor
|
229 |
+
normal_predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-8-1')
|
230 |
|
231 |
# Launch the app
|
232 |
demo.launch()
|
requirements.txt
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
|
|
7 |
torch==2.4.0
|
8 |
torchvision==0.19.0
|
9 |
pillow==10.4.0
|
|
|
1 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
2 |
+
huggingface_hub==0.25.0
|
3 |
+
diffusers==0.28.0
|
4 |
+
accelerate==1.2.1
|
5 |
+
kornia==0.8.0
|
6 |
+
timm==0.6.7
|
7 |
+
|
8 |
torch==2.4.0
|
9 |
torchvision==0.19.0
|
10 |
pillow==10.4.0
|