update

app.py

@@ -654,9 +654,16 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 # Launch the Gradio app
 if __name__ == "__main__":
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
-    pipeline.cuda()
+    if torch.cuda.is_available():
+        pipeline.cuda()
+        print("CUDA is available. Using GPU.")
+    else:
+        print("CUDA not available. Falling back to CPU.")
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:
         pass
-    demo.launch(show_error=True,show_api=True)
+    print(f"CUDA Available: {torch.cuda.is_available()}")
+    print(f"CUDA Version: {torch.version.cuda}")
+    print(f"Number of GPUs: {torch.cuda.device_count()}")
+    demo.launch(debug=True)