TRELLIS-JT

Runtime error

JeffreyXiang commited on Dec 6, 2024

Commit

41cbb4a

verified ·

1 Parent(s): 4634466

Fix rembg

Files changed (1) hide show

app.py CHANGED Viewed

@@ -250,4 +250,5 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
     pipeline.cuda()
     demo.launch()

 if __name__ == "__main__":
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
     pipeline.cuda()
+    pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     demo.launch()