feat: Init
This commit is contained in:
@@ -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"]
|
||||
@@ -0,0 +1,15 @@
|
||||
FROM python:3.12-slim
|
||||
|
||||
ENV PYTHONDONTWRITEBYTECODE=1
|
||||
ENV PYTHONUNBUFFERED=1
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# Только легкие зависимости для отображения интерфейса
|
||||
RUN pip install --no-cache-dir streamlit==1.32.0 requests pandas pillow
|
||||
|
||||
COPY src/ /app/src/
|
||||
|
||||
EXPOSE 8080
|
||||
|
||||
CMD ["streamlit", "run", "src/main.py", "--server.port", "8080", "--server.address", "0.0.0.0"]
|
||||
Reference in New Issue
Block a user