- TTS: xtts-v2 integration with voice cloning support
- Document: docling integration for PDF/DOCX/PPTX processing
- Memory Service: added /facts/upsert, /facts/{key}, /facts endpoints
- Added required dependencies (TTS, docling)
42 lines
689 B
Docker
42 lines
689 B
Docker
FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
|
|
|
|
WORKDIR /app
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
python3.11 \
|
|
python3-pip \
|
|
curl \
|
|
git \
|
|
libgl1-mesa-glx \
|
|
libglib2.0-0 \
|
|
poppler-utils \
|
|
tesseract-ocr \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install Python dependencies
|
|
RUN pip3 install --no-cache-dir \
|
|
docling \
|
|
docling-core \
|
|
torch \
|
|
torchvision \
|
|
transformers \
|
|
accelerate \
|
|
pillow \
|
|
fastapi \
|
|
uvicorn \
|
|
python-multipart \
|
|
pydantic \
|
|
httpx \
|
|
PyMuPDF \
|
|
pdf2image
|
|
|
|
# Copy service code
|
|
COPY . /app
|
|
|
|
# Expose port
|
|
EXPOSE 8003
|
|
|
|
# Run service
|
|
CMD ["python3", "main.py"]
|