feat: Init

This commit is contained in:
zin
2026-06-02 22:39:11 +00:00
parent 875616730b
commit 3850b15053
7 changed files with 210 additions and 120 deletions
+20
View File
@@ -0,0 +1,20 @@
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
WORKDIR /app
RUN apt-get update && apt-get install -y \
libglib2.0-0 libsm6 libxext6 libxrender-dev \
&& rm -rf /var/lib/apt/lists/*
# Устанавливаем зависимости для ML и API напрямую, чтобы не плодить requirements.txt
RUN pip install --no-cache-dir fastapi uvicorn timm scikit-learn pandas joblib python-multipart
COPY src/ /app/src/
EXPOSE 8000
# Запускаем FastAPI сервер
CMD ["uvicorn", "src.api:app", "--host", "0.0.0.0", "--port", "8000"]