Fix Dockerfile build errors and always use real model

Build Fixes:
- Add cmake and build-essential to system dependencies
- Install PyTorch before building C++ extensions (required)
- Build only mycuda and mycpp (skip optional kaolin)
- Use proper pip install commands for C++ extensions

Always Real Mode:
- Remove USE_HF_WEIGHTS and USE_REAL_MODEL env var checks
- Always download weights from gpue/foundationpose-weights
- Set USE_REAL_MODEL=true permanently in app.py
- Update UI to show 'Real FoundationPose with GPU'
- Update README to mention Docker GPU (not ZeroGPU)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

Dockerfile

@@ -6,15 +6,16 @@ ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
-# FoundationPose configuration
+# FoundationPose configuration - always use real model
 ENV FOUNDATIONPOSE_MODEL_REPO=gpue/foundationpose-weights
-ENV USE_HF_WEIGHTS=true
-ENV USE_REAL_MODEL=false
+ENV USE_REAL_MODEL=true
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     wget \
+    cmake \
+    build-essential \
     python3.10 \
     python3.10-dev \
     python3-pip \
@@ -36,27 +37,27 @@ RUN python3 -m pip install --upgrade pip
 # Set working directory
 WORKDIR /app
 
-# Install Python dependencies first (for better Docker layer caching)
+# Install Python dependencies first (PyTorch needed for building C++ extensions)
 COPY requirements.txt .
+RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cu118
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Clone FoundationPose repository
 RUN git clone https://github.com/NVlabs/FoundationPose.git /app/FoundationPose
 
-# Build FoundationPose C++ extensions
+# Build FoundationPose C++ extensions (skip kaolin - optional dependency)
 WORKDIR /app/FoundationPose
-RUN bash build_all.sh || echo "⚠️  Build completed with warnings"
+RUN cd bundlesdf/mycuda && pip install -e .
+RUN cd mycpp && python setup.py build_ext --inplace
 
 # Copy application files
 WORKDIR /app
-COPY app.py client.py estimator.py download_weights.py ./
+COPY app.py client.py estimator.py ./
 
-# Create weights directory
+# Create weights directory and download model weights
 RUN mkdir -p weights
-
-# Download weights if USE_HF_WEIGHTS=true (optional at build time)
-# Weights can also be downloaded at runtime
-RUN python3 download_weights.py || echo "⚠️  Weight download skipped"
+RUN python3 -c "from huggingface_hub import snapshot_download; \
+snapshot_download(repo_id='gpue/foundationpose-weights', local_dir='weights', repo_type='model')"
 
 # Expose Gradio port
 EXPOSE 7860

README.md

@@ -16,7 +16,7 @@ tags:
 
 # FoundationPose Inference Server
 
-This Hugging Face Space provides 6D object pose estimation using [FoundationPose](https://github.com/NVlabs/FoundationPose) with ZeroGPU support.
+This Hugging Face Space provides 6D object pose estimation using [FoundationPose](https://github.com/NVlabs/FoundationPose) with GPU support via Docker.
 
 ## Features
 

app.py

@@ -22,10 +22,10 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Check if running in real FoundationPose mode or placeholder mode
-USE_REAL_MODEL = os.environ.get("USE_REAL_MODEL", "false").lower() == "true"
+# Always use real FoundationPose model
+USE_REAL_MODEL = True
 
-logger.info(f"Starting in {'REAL' if USE_REAL_MODEL else 'PLACEHOLDER'} mode")
+logger.info("Starting in REAL mode with FoundationPose")
 
 
 class FoundationPoseInference:
@@ -282,7 +282,7 @@ with gr.Blocks(title="FoundationPose Inference", theme=gr.themes.Soft()) as demo
     gr.Markdown("# 🎯 FoundationPose 6D Object Pose Estimation")
 
     mode_indicator = gr.Markdown(
-        f"**Mode:** {'🟢 Real FoundationPose' if USE_REAL_MODEL else '🟡 Placeholder'}",
+        "**Mode:** 🟢 Real FoundationPose with GPU",
         elem_id="mode"
     )
 

download_weights.py

@@ -4,29 +4,23 @@
 import os
 from pathlib import Path
 
-# Check environment variables
+# Always download weights from model repository
 repo = os.environ.get('FOUNDATIONPOSE_MODEL_REPO', 'gpue/foundationpose-weights')
 token = os.environ.get('HF_TOKEN')
-use_hf = os.environ.get('USE_HF_WEIGHTS', 'false').lower() == 'true'
-use_real = os.environ.get('USE_REAL_MODEL', 'false').lower() == 'true'
 
-if use_hf and use_real:
-    print(f'Downloading weights from {repo}...')
+print(f'Downloading weights from {repo}...')
 
-    try:
-        from huggingface_hub import snapshot_download
+try:
+    from huggingface_hub import snapshot_download
 
-        snapshot_download(
-            repo_id=repo,
-            local_dir='weights',
-            token=token,
-            repo_type='model'
-        )
+    snapshot_download(
+        repo_id=repo,
+        local_dir='weights',
+        token=token,
+        repo_type='model'
+    )
 
-        print('✓ Weights downloaded successfully')
-    except Exception as e:
-        print(f'⚠️  Weight download failed: {e}')
-        print('Space will run in placeholder mode')
-else:
-    print('Placeholder mode - skipping weight download')
-    print('Set USE_REAL_MODEL=true and USE_HF_WEIGHTS=true to enable')
+    print('✓ Weights downloaded successfully')
+except Exception as e:
+    print(f'✗ Weight download failed: {e}')
+    raise