JeffreyXiang
commited on
Commit
•
c666caf
1
Parent(s):
41cbb4a
Fix
Browse files
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 |
-
|
|
|
|
|
|
|
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()
|