BarBar288 commited on
Commit
8132a50
·
verified ·
1 Parent(s): ff68386

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -9,7 +9,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
-
 
13
  # Set the environment variable for Hugging Face access token
14
  ARG ACC_TOKEN
15
  ENV AccToken=${ACC_TOKEN}
 
9
 
10
  # Install any needed packages specified in requirements.txt
11
  RUN pip install --no-cache-dir -r requirements.txt
12
+ # Allocate Resources
13
+ RUN docker run -p 7860:7860 --env ACC_TOKEN=your_huggingface_token --cpus=4 --memory=8g your-image-name
14
  # Set the environment variable for Hugging Face access token
15
  ARG ACC_TOKEN
16
  ENV AccToken=${ACC_TOKEN}