File size: 487 Bytes
7589132
 
ad29a66
c66f90e
4ca2511
 
 
3a9708b
6fda8a4
defbc57
3a9708b
ad29a66
3f32a55
ff2b03c
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
FROM nvcr.io/nvidia/pytorch:23.12-py3

WORKDIR /code

RUN pip install ultralytics
RUN pip install opencv-python==4.6.0.66
RUN pip install Pillow==10.3.0
RUN pip install uvicorn fastapi
RUN apt update && apt install fonts-dejavu -y
RUN wget https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-pose.pt

COPY . .

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

# curl -X POST "https://unfinity-yolov8-pose.hf.space/predict-image" -F "[email protected]"