Hugging Face's logo

Spaces:
Dragunflie-420
/
TRELLIS_3D
Running on Zero

App Files Community
JeffreyXiang commited on
Commit
41cbb4a
•
1 Parent(s): 4634466

Fix rembg

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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()