FROM dnth/ram-plus-image-tagging-gpu

# Set BentoML home to a writable directory
ENV BENTOML_HOME=/home/user/bentoml

# Create a non-root user
RUN useradd -m -u 1000 user

# Create BentoML home directory and set permissions
RUN mkdir -p /home/user/bentoml && chown -R user:user /home/user/bentoml

# Switch to non-root user
USER user