cxeep commited on
Commit
f96091f
·
verified ·
1 Parent(s): 199444f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +12 -10
Dockerfile CHANGED
@@ -8,15 +8,6 @@ RUN apt-get update && apt-get install -y \
8
 
9
  WORKDIR /code
10
 
11
- # Clone the repository
12
- RUN git clone https://github.com/Truoji/tengine-lite-yolov5s-tt100k .
13
-
14
- # Create build directory and build the project
15
- RUN mkdir build && cd build && cmake .. && make
16
-
17
- # Set environment variable
18
- ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/code/third_party/
19
-
20
  # Copy requirements and install dependencies
21
  COPY ./requirements.txt /code/requirements.txt
22
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
@@ -34,8 +25,19 @@ ENV HOME=/home/user \
34
  # Set the working directory to the user's home directory
35
  WORKDIR $HOME/app
36
 
 
 
 
 
 
 
 
 
 
 
 
37
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
38
  COPY --chown=user . $HOME/app
39
 
40
  # Run the project
41
- CMD ["python", "gradio.app.py"]
 
8
 
9
  WORKDIR /code
10
 
 
 
 
 
 
 
 
 
 
11
  # Copy requirements and install dependencies
12
  COPY ./requirements.txt /code/requirements.txt
13
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
25
  # Set the working directory to the user's home directory
26
  WORKDIR $HOME/app
27
 
28
+ # Clone and build the project
29
+ RUN git clone https://github.com/Truoji/tengine-lite-yolov5s-tt100k && \
30
+ cd tengine-lite-yolov5s-tt100k && \
31
+ mkdir build && \
32
+ cd build && \
33
+ cmake .. && \
34
+ make
35
+
36
+ # Set LD_LIBRARY_PATH to include third_party libraries
37
+ ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/app/tengine-lite-yolov5s-tt100k/third_party/
38
+
39
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
40
  COPY --chown=user . $HOME/app
41
 
42
  # Run the project
43
+ CMD ["python", "gradio.app.py"]