AIcoder35235 commited on
Commit
c4bd273
·
verified ·
1 Parent(s): a707ddc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -6,9 +6,9 @@ WORKDIR /app
6
 
7
  # 3. Install PyTorch (CPU version to save massive space/RAM) and required libraries
8
  RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
9
-
10
  # 4. Install API and networking dependencies (requests added here!)
11
- RUN pip install --no-cache-dir fastapi uvicorn transformers pillow pydantic requests
12
 
13
  # 5. THE CHEAT CODE: Pre-download the heavy model weights during the build phase
14
  RUN python -c "from transformers import pipeline; pipeline('image-classification', model='dima806/chest_xray_pneumonia_detection')"
 
6
 
7
  # 3. Install PyTorch (CPU version to save massive space/RAM) and required libraries
8
  RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
9
+ RUN pip install --no-cache-dir fastapi uvicorn transformers pillow pydantic requests beautifulsoup4
10
  # 4. Install API and networking dependencies (requests added here!)
11
+
12
 
13
  # 5. THE CHEAT CODE: Pre-download the heavy model weights during the build phase
14
  RUN python -c "from transformers import pipeline; pipeline('image-classification', model='dima806/chest_xray_pneumonia_detection')"