JeffreyXiang commited on
Commit
c666caf
1 Parent(s): 41cbb4a
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -250,5 +250,8 @@ with gr.Blocks() as demo:
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()
 
250
  if __name__ == "__main__":
251
  pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
252
  pipeline.cuda()
253
+ try:
254
+ pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
255
+ except:
256
+ pass
257
  demo.launch()