IamVicky111 commited on
Commit
52115ed
·
verified ·
1 Parent(s): 2b56608

Update app.Dockerfile

Browse files
Files changed (1) hide show
  1. app.Dockerfile +3 -3
app.Dockerfile CHANGED
@@ -7,11 +7,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
7
  python3-dev \
8
  wget \
9
  curl \
10
- libnvidia-cuda-dev # Adapt based on your CUDA version
11
- # Add other necessary dependencies here
12
 
13
- RUN pip install --no-cache-dir torch torchvision torchaudio
14
  RUN pip install --no-cache-dir transformers
 
 
15
 
16
  WORKDIR /app
17
 
 
7
  python3-dev \
8
  wget \
9
  curl \
10
+ libnvidia-cuda-dev
 
11
 
 
12
  RUN pip install --no-cache-dir transformers
13
+ RUN pip install --no-cache-dir torch torchvision torchaudio
14
+
15
 
16
  WORKDIR /app
17