Update Dockerfile

Dockerfile

@@ -34,6 +34,21 @@ RUN mkdir -p /workspaces/empty/build/bin && \
 # Make sure binaries are executable
 RUN chmod +x /workspaces/empty/build/bin/*
 
+# Create cache directory and set permissions
+RUN mkdir -p /.cache/llama.cpp && \
+    chmod -R 777 /.cache
+
+# Create a non-root user for running the application
+RUN groupadd -r llama && useradd -r -g llama llama && \
+    mkdir -p /home/llama/.cache/llama.cpp && \
+    chown -R llama:llama /home/llama
+
+# Set environment variable for cache directory
+ENV XDG_CACHE_HOME=/home/llama/.cache
+
+# Switch to non-root user
+USER llama
+
 # Step 4 & 5: Set Environment Variables for PATH and LD_LIBRARY_PATH
 ENV PATH="/workspaces/empty/build/bin:${PATH}"
 ENV LD_LIBRARY_PATH="/workspaces/empty/build/bin:${LD_LIBRARY_PATH}"