mingyang91 commited on
Commit
a9311a8
·
verified ·
1 Parent(s): 2b27506
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,5 +1,6 @@
1
  FROM python:3.11-bookworm
2
  ARG HF_TOKEN
 
3
  RUN useradd -m -u 1000 user
4
  USER user
5
  # Set home to the user's home directory
@@ -12,5 +13,5 @@ RUN mkdir $HOME/.cache
12
  RUN chown -R user $HOME/.cache
13
 
14
  RUN pip install -r requirements.txt
15
- RUN HF_TOKEN=$HF_TOKEN python yolo_fire.py
16
  ENTRYPOINT ["streamlit", "run", "demo.py"]
 
1
  FROM python:3.11-bookworm
2
  ARG HF_TOKEN
3
+ ENV HF_TOKEN=$HF_TOKEN
4
  RUN useradd -m -u 1000 user
5
  USER user
6
  # Set home to the user's home directory
 
13
  RUN chown -R user $HOME/.cache
14
 
15
  RUN pip install -r requirements.txt
16
+ RUN python yolo_fire.py
17
  ENTRYPOINT ["streamlit", "run", "demo.py"]