Update Dockerfile

Dockerfile

@@ -1,19 +1,25 @@
 FROM h2oairelease/h2oai-llmstudio-app:latest
 
-# Volta para root temporariamente para criar diretórios
 USER root
 
-RUN mkdir -p /home/llmstudio/mount && \
-    chmod 777 /home/llmstudio/mount
+RUN mkdir -p /home/user/mount && \
+    mkdir -p /home/user/.triton && \
+    chown -R 1000:1000 /home/user
 
-ENV H2O_WAVE_MAX_REQUEST_SIZE=25MB \
+USER 1000
+
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    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
+    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
 
+WORKDIR /home/user
 EXPOSE 10101
 
-# Volta para o usuário padrão da imagem (llmstudio)
-USER llmstudio
-
 CMD ["wave", "run", "llm_studio.app"]