CyberSecChatbot / Dockerfile
Andrew McCracken
Disable RAG for faster inference
2a55dc3
raw
history blame
808 Bytes
# Use pre-built image from Docker Hub
# Build this image locally with: ./build-and-push.sh
FROM techdaskalos/cybersecchatbot:latest
# Environment variables (already set in base image, but can override)
ENV PYTHONUNBUFFERED=1
ENV MODEL_REPO=daskalos-apps/phi4-cybersec-Q4_K_M
ENV MODEL_FILENAME=phi4-mini-instruct-Q4_K_M.gguf
ENV USE_RAG=false
ENV CACHE_ENABLED=true
# Set Hugging Face cache to /data for persistence and write permissions
ENV HF_HOME=/data/huggingface
# Ensure all required directories exist and are writable
RUN mkdir -p /data /app/models /app/knowledge_db /data/huggingface/hub /data/huggingface/transformers && \
chmod -R 777 /data /app/models /app/knowledge_db
# Copy test interface (needed for /test endpoint)
COPY test_interface.html /app/
EXPOSE 8000
CMD ["python", "main.py"]