Update Dockerfile

Dockerfile

@@ -7,6 +7,9 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Set environment variable for HF cache
+ENV HF_HOME=/tmp/huggingface_cache
+
 # Clone llama.cpp
 RUN git clone https://github.com/ggerganov/llama.cpp /app/llama.cpp
 
@@ -22,5 +25,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy your app code
 COPY . .
 
+# Ensure the cache directory exists
+RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
+
 # Set the entrypoint
-CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
+CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]