Spaces:
Sleeping
Sleeping
File size: 380 Bytes
ab794cc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
WORKDIR /app
COPY requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir -r /app/requirements.txt
COPY app /app/app
COPY src /app/src
COPY artifacts /app/artifacts
ENV MODEL_PATH=/app/artifacts/model_best.pt
ENV THRESHOLD=0.5
EXPOSE 7860
CMD ["uvicorn", "app.api:app", "--host", "0.0.0.0", "--port", "7860"]
|