BarBar288 commited on
Commit
2bdcc3d
·
verified ·
1 Parent(s): 0ff3338

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -2
Dockerfile CHANGED
@@ -7,10 +7,15 @@ WORKDIR /home/user/app
7
  # Copy the current directory contents into the container at /home/user/app
8
  COPY . /home/user/app
9
 
 
 
 
 
 
 
10
  # Install any needed packages specified in requirements.txt
11
  RUN pip install --no-cache-dir -r requirements.txt
12
- # Allocate Resources
13
- docker run -p 7860:7860 --env ACC_TOKEN=your_huggingface_token --cpus=32 --memory=200g --gpus=10 your-image-name
14
  # Set the environment variable for Hugging Face access token
15
  ARG ACC_TOKEN
16
  ENV AccToken=${ACC_TOKEN}
 
7
  # Copy the current directory contents into the container at /home/user/app
8
  COPY . /home/user/app
9
 
10
+ # Install necessary system dependencies
11
+ RUN apt-get update && apt-get install -y \
12
+ build-essential \
13
+ libgl1-mesa-glx \
14
+ && rm -rf /var/lib/apt/lists/*
15
+
16
  # Install any needed packages specified in requirements.txt
17
  RUN pip install --no-cache-dir -r requirements.txt
18
+
 
19
  # Set the environment variable for Hugging Face access token
20
  ARG ACC_TOKEN
21
  ENV AccToken=${ACC_TOKEN}