Update Dockerfile

Dockerfile

@@ -2,23 +2,22 @@ FROM h2oairelease/h2oai-llmstudio-app:latest
 
 USER root
 
-RUN mkdir -p /home/user/mount && \
-    mkdir -p /home/user/.triton && \
-    chown -R 1000:1000 /home/user && \
+RUN mkdir -p /workspace/mount && \
+    chmod -R 777 /workspace/mount && \
     chmod -R 777 /workspace
 
 USER 1000
 
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
+    H2O_LLM_STUDIO_WORKDIR=/workspace/mount \
+    HF_HOME=/workspace/mount/huggingface \
+    TRITON_CACHE_DIR=/workspace/mount/.triton/cache \
+    H2O_WAVE_DATA_DIR=/workspace/mount/wave_data \
     H2O_WAVE_MAX_REQUEST_SIZE=25MB \
     H2O_WAVE_NO_LOG=True \
-    H2O_WAVE_PRIVATE_DIR="/download/@output/download" \
-    H2O_WAVE_ADDRESS=http://0.0.0.0:10101 \
-    TRITON_CACHE_DIR=/home/user/.triton \
-    TRITON_HOME=/home/user/.triton \
-    HF_HOME=/home/user/.cache/huggingface \
-    TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+    H2O_WAVE_PRIVATE_DIR="/download/@/workspace/mount/output/download" \
+    H2O_WAVE_ADDRESS=http://0.0.0.0:10101
 
 WORKDIR /workspace