Update Dockerfile

Dockerfile

@@ -1,9 +1,10 @@
 FROM python:3.12.9-slim-bookworm
 WORKDIR /app
+ENV OLLAMA_DIR=/app/.ollama
 COPY requirements.txt .
 RUN apt update && apt install -y curl git
 RUN pip install --no-cache-dir -r requirements.txt
-RUN mkdir -p /app/logs && chmod 777 /app/logs
+RUN mkdir -p /app/logs /app/.ollama && chmod -R 777 /app/.ollama /app/logs
 RUN python -c "from transformers import AutoModelForSeq2SeqLM; AutoModelForSeq2SeqLM.from_pretrained('facebook/bart-large-cnn')"
 RUN curl -fsSL https://ollama.com/install.sh -o install.sh && chmod +x install.sh && ./install.sh && rm install.sh
 COPY . .