upd requirements and dockerfile

Dockerfile

@@ -1,12 +1,20 @@
 FROM nvcr.io/nvidia/pytorch:23.12-py3
 
+# YOLOv8
 RUN pip install ultralytics
-RUN pip install streamlit
 RUN pip install opencv-python==4.6.0.66
 RUN pip install Pillow==10.3.0
 
+# FastAPI
+RUN pip install fastapi uvicorn
+
+# Streamlit
+RUN pip install streamlit
+
+# Fonts
 RUN apt update && apt install fonts-dejavu -y
 
+# Download model weights
 WORKDIR /app
 RUN wget https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-pose.pt -O /app/yolov8l-pose.pt
 

requirements.txt

@@ -0,0 +1,7 @@
+torch
+ultralytics
+opencv-python==4.6.0.66
+Pillow==10.3.0
+fastapi
+uvicorn
+streamlit