updated docker

Dockerfile

@@ -1,11 +1,13 @@
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
-# System-level environment variables
+# Set environment variables
 ENV HOST 0.0.0.0
 ENV HOME /home/user
 ENV PATH $HOME/.local/bin:$PATH
+ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
+ENV TMPDIR=/home/user/tmp
 
-# Install system dependencies for Python and build tools
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     python3 python3-pip build-essential git \
     && rm -rf /var/lib/apt/lists/*
@@ -17,12 +19,12 @@ RUN useradd -m -u 1000 user
 RUN pip install --no-cache-dir --upgrade pip \
     && pip install --no-cache-dir poetry
 
+# Create temporary and cache directories for Poetry
+RUN mkdir -p /home/user/tmp && chown -R user:user /home/user/tmp
+
 # Switch to non-root user
 USER user
 
-# Ensure Poetry works for non-root user
-ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
-
 # Set the working directory
 WORKDIR /home/user/app
 
@@ -40,4 +42,4 @@ COPY --chown=user . /home/user/app
 EXPOSE 8501
 
 # Run the Streamlit app
-CMD ["streamlit", "run", "app.py", "--server.headless=true", "--server.enableCORS=false", "--server.enableXsrfProtection=false", "--server.fileWatcherType=none"]
+CMD ["streamlit", "run", "app.py", "--server.headless=true", "--server.enableCORS=false", "--server.enableXsrfProtection=false", "--server.fileWatcherType=none"]