Update Dockerfile

Dockerfile

@@ -1,16 +1,31 @@
-FROM ubuntu:latest
+FROM python:3.10
 
-ENV DEBIAN_FRONTEND=noninteractive
+# Install necessary libraries
+RUN apt-get update && apt-get install -y \
+    git \
+    cmake \
+    libopencv-dev
 
-# Install system dependencies
-RUN apt-get update && \
-    apt-get install -y libopencv-dev git cmake build-essential python3 python3-pip && \
-    rm -rf /var/lib/apt/lists/*
+WORKDIR /code
 
-# Set up a new user named "user" with user ID 1000 and create the necessary directories
-RUN useradd -m -u 1000 user && \
-    mkdir -p /home/user/app && \
-    chown -R user:user /home/user/app
+# Clone the repository
+RUN git clone https://github.com/Truoji/tengine-lite-yolov5s-tt100k .
+
+# Create build directory and build the project
+RUN mkdir build && cd build && cmake .. && make
+
+# Set environment variable
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/code/third_party/
+
+# Make yolov5s-tt100k executable
+RUN chmod +x ./yolov5s-tt100k
+
+# Copy requirements and install dependencies
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
 
 # Switch to the "user" user
 USER user
@@ -19,32 +34,11 @@ USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 
-# Set the working directory to the user's app directory
+# Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
-# Clone the GitHub repository as the "user"
-RUN git clone https://github.com/Truoji/tengine-lite-yolov5s-tt100k.git
-
-# Set the working directory to the cloned repository
-WORKDIR $HOME/app/tengine-lite-yolov5s-tt100k
-
-# Build the project
-RUN mkdir build && \
-    cd build && \
-    cmake .. && \
-    make
-
-# Set environment variables
-ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${HOME}/app/tengine-lite-yolov5s-tt100k/third_party/"
-
-# Install Python packages as the "user"
-RUN pip3 install --user gradio
-
-# Copy the gradio.app.py script into the container at $HOME/app/tengine-lite-yolov5s-tt100k
-COPY --chown=user gradio.app.py ${HOME}/app/tengine-lite-yolov5s-tt100k/gradio.app.py
-
-# Expose the port on which the web app will run
-EXPOSE 7860
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
 
-# Run gradio.app.py when the container launches
-CMD ["python3", "gradio.app.py"]
+# Run the project
+CMD ["python", "gradio.app.py"]