menhso / Dockerfile
chihq
Add Gradio UI at /ui for testing
2ee1270
Raw
History Blame Contribute Delete
678 Bytes
FROM ultralytics/ultralytics:latest-cpu
USER root
RUN apt-get update && apt-get install -y --no-install-recommends \
libsm6 libxext6 libxrender1 \
&& rm -rf /var/lib/apt/lists/*
RUN pip install --no-cache-dir fastapi uvicorn mediapipe pillow supervision gradio
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
WORKDIR /home/user/app
COPY --chown=user inference_server.py .
COPY --chown=user models/ ./models/
ENV WEIGHTS_PATH=/home/user/app/models/weights.onnx
ENV MEDIAPIPE_MODEL_PATH=/home/user/app/models/hand_landmarker.task
EXPOSE 7860
CMD ["uvicorn", "inference_server:app", "--host", "0.0.0.0", "--port", "7860"]