Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -20,7 +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 requests psutil termcolor
|
24 |
|
25 |
# Copy the current directory contents into the container at /app
|
26 |
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 requests psutil termcolor sk-learn
|
24 |
|
25 |
# Copy the current directory contents into the container at /app
|
26 |
COPY . /app
|