Commit
•
41cbb4a
1
Parent(s):
4634466
Fix rembg
Browse files
app.py
CHANGED
|
@@ -250,4 +250,5 @@ with gr.Blocks() as demo:
|
|
| 250 |
if __name__ == "__main__":
|
| 251 |
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
| 252 |
pipeline.cuda()
|
|
|
|
| 253 |
demo.launch()
|
|
|
|
| 250 |
if __name__ == "__main__":
|
| 251 |
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
| 252 |
pipeline.cuda()
|
| 253 |
+
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|
| 254 |
demo.launch()
|