Update Dockerfile

Dockerfile

@@ -2,12 +2,15 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
 # Install Python 3.10 and system dependencies
 RUN apt-get update && \
-    apt-get install -y python3.10 python3.10-dev python3-pip libgl1-mesa-glx libglib2.0-0 ffmpeg && \
+    apt-get install -y python3.10 python3.10-dev python3-pip git libgl1-mesa-glx libglib2.0-0 ffmpeg && \
     ln -s /usr/bin/python3.10 /usr/bin/python
 
 # Set working directory
 WORKDIR /app
 
+# Clone YOLOv7 repository
+RUN git clone https://github.com/WongKinYiu/yolov7.git
+
 # Copy all app files into container
 COPY . /app
 
@@ -17,3 +20,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 # Run Streamlit app on container start
 CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
+