Spaces:
Running
on
Zero
Running
on
Zero
kushbhargav
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -405,7 +405,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
|
|
405 |
|
406 |
# Launch the Gradio app
|
407 |
if __name__ == "__main__":
|
408 |
-
pipeline = TrellisImageTo3DPipeline.from_pretrained("
|
409 |
pipeline.cuda()
|
410 |
try:
|
411 |
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|
|
|
405 |
|
406 |
# Launch the Gradio app
|
407 |
if __name__ == "__main__":
|
408 |
+
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
409 |
pipeline.cuda()
|
410 |
try:
|
411 |
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|