fix: Update app

README.md

@@ -1,5 +1,5 @@
 ---
-title: TRELLIS
+title: Hi3DGen
 emoji: 🏢
 colorFrom: indigo
 colorTo: blue
@@ -8,9 +8,9 @@ sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 license: mit
-short_description: Scalable and Versatile 3D Generation from images
+short_description: High-fidelity 3D Geometry Generation from images
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
-Paper: https://huggingface.co/papers/2412.01506
+Paper: 

app.py

@@ -223,9 +223,10 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
 if __name__ == "__main__":
     # Initialize pipeline
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
+    pipeline.cuda()
     
     # Initialize normal predictor
-    # normal_predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-8-1')
+    normal_predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-8-1')
     
     # Launch the app
     demo.launch()

requirements.txt

@@ -1,9 +1,10 @@
 --extra-index-url https://download.pytorch.org/whl/cu121
-#huggingface_hub==0.25.0
-#diffusers==0.28.0
-#accelerate==1.2.1
-#kornia==0.8.0
-#timm==0.6.7
+huggingface_hub==0.25.0
+diffusers==0.28.0
+accelerate==1.2.1
+kornia==0.8.0
+timm==0.6.7
+
 torch==2.4.0
 torchvision==0.19.0
 pillow==10.4.0