Spaces:
Sleeping
Sleeping
| FROM python:3.11-slim | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| libgl1 \ | |
| libglib2.0-0 \ | |
| && rm -rf /var/lib/apt/lists/* | |
| WORKDIR /code | |
| COPY app/requirements.txt /code/requirements.txt | |
| RUN pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu \ | |
| -r /code/requirements.txt | |
| ENV HF_HOME=/code/.hf-cache | |
| ENV PYTHONUNBUFFERED=1 | |
| RUN python -c "from transformers import AutoImageProcessor, AutoModelForImageClassification; \ | |
| AutoImageProcessor.from_pretrained('microsoft/swinv2-base-patch4-window16-256', cache_dir='/code/.hf-cache'); \ | |
| AutoModelForImageClassification.from_pretrained('microsoft/swinv2-base-patch4-window16-256', cache_dir='/code/.hf-cache')" | |
| COPY model.safetensors /code/model.safetensors | |
| COPY app /code/app | |
| EXPOSE 7860 | |
| CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"] | |