Update Dockerfile

Dockerfile

@@ -10,17 +10,19 @@ ENV OPENAI_API_KEY=EMPTY
 ENV GRADIO_SERVER_NAME=0.0.0.0
 ENV GRADIO_SERVER_PORT=7860
 
-ENV VLLM_PORT=8000
-ENV MAX_MODEL_LEN=8192
+ENV VLLM_PORT=8001
+ENV MAX_MODEL_LEN=18000
 ENV GPU_MEMORY_UTILIZATION=0.90
 
-ENV NUEXTRACT_MAX_TOKENS=5000
+ENV NUEXTRACT_MAX_TOKENS=15000
 ENV NUEXTRACT_EXAMPLE_DIR=/home/user/app/examples
 ENV NUEXTRACT_ASSETS_DIR=/home/user/app/assets
 
-ENV HF_HOME=/data/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/data/.cache/huggingface
-ENV VLLM_CACHE_ROOT=/data/.cache/vllm
+ENV HF_HOME=/home/user/.cache/huggingface
+ENV HUGGINGFACE_HUB_CACHE=/home/user/.cache/huggingface/hub
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+ENV VLLM_CACHE_ROOT=/home/user/.cache/vllm
+ENV XDG_CACHE_HOME=/home/user/.cache
 
 RUN pip install --no-cache-dir \
     gradio \
@@ -36,8 +38,12 @@ COPY --chown=user:user start.sh /home/user/app/start.sh
 COPY --chown=user:user assets /home/user/app/assets
 COPY --chown=user:user examples /home/user/app/examples
 
-RUN chown -R user:user /home/user/app && \
-    chmod +x /home/user/app/start.sh
+RUN mkdir -p \
+      /home/user/.cache/huggingface \
+      /home/user/.cache/huggingface/hub \
+      /home/user/.cache/vllm \
+    && chown -R user:user /home/user \
+    && chmod +x /home/user/app/start.sh
 
 USER user