Update Dockerfile

Dockerfile

@@ -3,6 +3,10 @@ FROM ubuntu:22.04
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV HOME=/root
+# CUDA paths for when running on GPU (RunPod)
+ENV CUDA_HOME=/usr/local/cuda
+ENV PATH=${CUDA_HOME}/bin:${PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
 # Install dependencies, add deadsnakes PPA for Python 3.12
 RUN apt-get update && \
@@ -39,7 +43,6 @@ RUN npm install -g @anthropic-ai/claude-code @anthropic-ai/dxt
 RUN mkdir -p /workspace
 
 # Copy requirements file (if it exists)
-# This goes BEFORE copying the rest of the code for better caching
 COPY requirements.txt* /workspace/
 
 # Install Python packages if requirements.txt exists