3v324v23 commited on
Commit
9bb25d7
·
1 Parent(s): c776eff

deploy: add curl for health checks and torchvision for cleaner logs

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. requirements.txt +2 -0
Dockerfile CHANGED
@@ -16,7 +16,7 @@ FROM python:3.11-slim
16
  WORKDIR /app
17
 
18
  # Install runtime dependencies and Nginx
19
- RUN apt-get update && apt-get install -y nginx && rm -rf /var/lib/apt/lists/*
20
  COPY requirements.txt .
21
  RUN pip install --no-cache-dir -r requirements.txt
22
 
@@ -33,3 +33,4 @@ EXPOSE 7860
33
 
34
  # Launch all services via entrypoint
35
  CMD ["./entrypoint.sh"]
 
 
16
  WORKDIR /app
17
 
18
  # Install runtime dependencies and Nginx
19
+ RUN apt-get update && apt-get install -y nginx curl && rm -rf /var/lib/apt/lists/*
20
  COPY requirements.txt .
21
  RUN pip install --no-cache-dir -r requirements.txt
22
 
 
33
 
34
  # Launch all services via entrypoint
35
  CMD ["./entrypoint.sh"]
36
+
requirements.txt CHANGED
@@ -1,9 +1,11 @@
1
  # Core ML
2
  torch>=2.1.0
 
3
  transformers>=4.35.0
4
  captum>=0.6.0
5
  torchdiffeq>=0.2.3
6
 
 
7
  # Backend & API
8
  fastapi>=0.104.0
9
  uvicorn>=0.24.0
 
1
  # Core ML
2
  torch>=2.1.0
3
+ torchvision>=0.16.0
4
  transformers>=4.35.0
5
  captum>=0.6.0
6
  torchdiffeq>=0.2.3
7
 
8
+
9
  # Backend & API
10
  fastapi>=0.104.0
11
  uvicorn>=0.24.0