dockerfile gpu

2 years ago · 9c9165aed6
parent 89b90b4eed
commit 9c9165aed6
1 changed files with 5 additions and 1 deletions
--- a/infra/Docker/Dockerfile.gpu
+++ b/infra/Docker/Dockerfile.gpu
@ -25,5 +25,9 @@ RUN python3.10 -m pip install --upgrade pip && \
 # Copy application files
 COPY . .

+
+# Expose port
+# EXPOSE ${EVAL_PORT}
+
 # Run the application
-CMD ["python3.10", "-m", "api.main", "serve"]
+CMD ["uvicorn", "api.app:app", "--host", "0.0.0.0", "--port", "8000"]