gensym commited on
Commit
915ce08
·
1 Parent(s): a8e79b5
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -276,6 +276,10 @@ cleanup_thread.start()
276
  @spaces.GPU
277
  def preload():
278
  pipeline.cuda()
 
 
 
 
279
 
280
 
281
  # Launch the Gradio app
@@ -283,8 +287,4 @@ if __name__ == "__main__":
283
  global pipeline
284
  pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
285
  preload()
286
- try:
287
- pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
288
- except:
289
- pass
290
  demo.launch()
 
276
  @spaces.GPU
277
  def preload():
278
  pipeline.cuda()
279
+ try:
280
+ pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
281
+ except:
282
+ pass
283
 
284
 
285
  # Launch the Gradio app
 
287
  global pipeline
288
  pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
289
  preload()
 
 
 
 
290
  demo.launch()