Update Dockerfile
Browse files- Dockerfile +1 -2
Dockerfile
CHANGED
|
@@ -20,8 +20,7 @@ WORKDIR /app
|
|
| 20 |
# Install Python dependencies
|
| 21 |
RUN pip install --upgrade pip
|
| 22 |
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 23 |
-
RUN pip install transformers accelerate
|
| 24 |
-
RUN pip install -r requirements.txt
|
| 25 |
|
| 26 |
# Copy the current directory contents into the container at /app
|
| 27 |
COPY . /app
|
|
|
|
| 20 |
# Install Python dependencies
|
| 21 |
RUN pip install --upgrade pip
|
| 22 |
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 23 |
+
RUN pip install transformers accelerate fastapi uvicorn requests psutil
|
|
|
|
| 24 |
|
| 25 |
# Copy the current directory contents into the container at /app
|
| 26 |
COPY . /app
|