|
|
|
|
|
|
|
|
|
|
|
FROM python:3.10-slim |
|
|
|
|
|
WORKDIR /app |
|
|
|
|
|
|
|
|
RUN apt-get update && apt-get install -y \ |
|
|
build-essential \ |
|
|
curl \ |
|
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
|
|
|
COPY requirements.txt . |
|
|
RUN pip install --no-cache-dir -r requirements.txt |
|
|
|
|
|
|
|
|
RUN mkdir -p /app/models /app/engine /app/utils |
|
|
|
|
|
|
|
|
COPY app.py . |
|
|
COPY engine/ ./engine/ |
|
|
COPY utils/ ./utils/ |
|
|
|
|
|
|
|
|
RUN python -c "from huggingface_hub import hf_hub_download; \ |
|
|
hf_hub_download( \ |
|
|
repo_id='GambitFlow/Synapse-Base', \ |
|
|
filename='synapse_base.onnx', \ |
|
|
local_dir='/app/models', \ |
|
|
local_dir_use_symlinks=False \ |
|
|
)" |
|
|
|
|
|
|
|
|
RUN ls -lh /app/models/synapse_base.onnx |
|
|
|
|
|
|
|
|
EXPOSE 7860 |
|
|
|
|
|
|
|
|
ENV PYTHONUNBUFFERED=1 |
|
|
ENV OMP_NUM_THREADS=2 |
|
|
ENV MKL_NUM_THREADS=2 |
|
|
ENV OPENBLAS_NUM_THREADS=2 |
|
|
|
|
|
|
|
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \ |
|
|
CMD curl -f http://localhost:7860/health || exit 1 |
|
|
|
|
|
|
|
|
CMD ["python", "app.py"] |