Update app.py
Browse files
app.py
CHANGED
|
@@ -421,13 +421,10 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
|
|
| 421 |
# Launch the Gradio app
|
| 422 |
if __name__ == "__main__":
|
| 423 |
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
| 424 |
-
|
| 425 |
|
| 426 |
pipeline.cuda()
|
| 427 |
-
|
| 428 |
-
# CPU 환경일 때 에러가 나지 않도록 임시 배치 (단, CPU는 TRELLIS 구동이 매우 느리거나 지원되지 않을 수 있음)
|
| 429 |
-
|
| 430 |
-
pipeline.cpu()
|
| 431 |
|
| 432 |
try:
|
| 433 |
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|
|
|
|
| 421 |
# Launch the Gradio app
|
| 422 |
if __name__ == "__main__":
|
| 423 |
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
| 424 |
+
|
| 425 |
|
| 426 |
pipeline.cuda()
|
| 427 |
+
|
|
|
|
|
|
|
|
|
|
| 428 |
|
| 429 |
try:
|
| 430 |
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|