Update Docker

Author: Trojan3877

Docker

@@ -1,10 +1,17 @@
-FROM python:3.10
+FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
 
 WORKDIR /app
 
+RUN apt-get update && apt-get install -y \
+    python3 python3-pip
+
 COPY requirements.txt .
-RUN pip install -r requirements.txt
+
+RUN pip3 install --upgrade pip
+RUN pip3 install -r requirements.txt
 
 COPY . .
 
-CMD ["uvicorn","app.infer_api:app","--host","0.0.0.0","--port","8000"]
+EXPOSE 8000
+
+CMD ["uvicorn", "api.main:app", "--host", "0.0.0.0", "--port", "8000"]