add reqs

Dockerfile

@@ -2,6 +2,9 @@ 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
 
 COPY . .

main.py

@@ -1,4 +1,14 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+import io
+import base64
+from ultralytics import YOLO
+import torch
+from PIL import Image, ImageOps
+
+import utils
+from drawing import draw_keypoints
+
 
 app = FastAPI()