Fix: Make entrypoint.sh handle missing nvidia-smi gracefully and switch to CPU hardware

entrypoint.sh

@@ -6,7 +6,19 @@ source /opt/mineru_venv/bin/activate
 
 # Display GPU information
 echo "Checking NVIDIA GPU status:"
-nvidia-smi
+if command -v nvidia-smi &> /dev/null; then
+    nvidia-smi
+else
+    echo "nvidia-smi command not found. GPU may not be available or drivers not installed."
+    echo "Continuing with CPU mode..."
+    
+    # Update the magic-pdf config to use CPU mode if GPU is not available
+    CONFIG_DIR="$HOME/.config/magic_pdf"
+    mkdir -p "$CONFIG_DIR"
+    if [ -f "$CONFIG_DIR/magic-pdf.json" ]; then
+        sed -i 's/"device-mode": "gpu"/"device-mode": "cpu"/g' "$CONFIG_DIR/magic-pdf.json" || true
+    fi
+fi
 
 # Display MinerU version
 echo "MinerU version:"
@@ -57,9 +69,16 @@ CONFIG_DIR="$HOME/.config/magic_pdf"
 mkdir -p "$CONFIG_DIR"
 if [ ! -f "$CONFIG_DIR/magic-pdf.json" ]; then
     echo "Creating magic-pdf.json configuration file..."
+    # Check if GPU is available to determine default mode
+    if command -v nvidia-smi &> /dev/null; then
+        DEVICE_MODE="gpu"
+    else
+        DEVICE_MODE="cpu"
+    fi
+    
     cat > "$CONFIG_DIR/magic-pdf.json" << EOF
 {
-    "device-mode": "gpu",
+    "device-mode": "$DEVICE_MODE",
     "layout-config": {
         "model": "doclayout_yolo",
         "enable": true

space_config.json

@@ -1,4 +1,4 @@
 {
     "runtime": "docker",
-    "hardware": "nvidia-l4-1x-16gb"
+    "hardware": "cpu"
 }