medsam2_sep490 / Dockerfile
Anhsapper's picture
Upload 9 files
4dfb638 verified
raw
history blame contribute delete
778 Bytes
FROM python:3.12-slim
# Cài đặt các thư viện hệ thống cần thiết
RUN apt-get update && apt-get install -y \
libgomp1 \
libgl1 \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
RUN pip install --upgrade pip
# Copy và cài đặt các python dependencies khác
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Tải sẵn model về Image để khởi động nhanh hơn
# RUN python -c "from transformers import SamModel, SamProcessor; \
# model_name = 'wanglab/medsam-vit-base'; \
# SamModel.from_pretrained(model_name); \
# SamProcessor.from_pretrained(model_name)"
COPY . .
EXPOSE 7860
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]