Stable-X commited on
Commit
1ab5d3a
·
1 Parent(s): 168b252

fix: Update app

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. app.py +2 -1
  3. requirements.txt +6 -5
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: TRELLIS
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: Scalable and Versatile 3D Generation from images
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
15
 
16
- Paper: https://huggingface.co/papers/2412.01506
 
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
- # normal_predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-8-1')
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
- #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
  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