Update Dockerfile

Dockerfile

@@ -1,26 +1,40 @@
 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 git libgl1-mesa-glx libglib2.0-0 ffmpeg && \
-    ln -s /usr/bin/python3.10 /usr/bin/python
+# Use the official Python image as the base
+FROM python:3.10-slim
+
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
 
 # Set working directory
 WORKDIR /app
 
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
 # Clone YOLOv7 repository
 RUN git clone https://github.com/WongKinYiu/yolov7.git
 
-# Copy all app files into container
-COPY . /app
-
-# Upgrade pip and install Python dependencies
-RUN pip install --upgrade pip
-RUN pip install --no-cache-dir -r requirements.txt
+# Copy application files
+COPY . .
 
-# Run Streamlit app on container start
-CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
+# Modify torch.load to set weights_only=False
+RUN sed -i 's/torch.load(\(.*\))/torch.load(\1, weights_only=False)/g' yolov7/models/experimental.py
 
+# Expose the port Streamlit runs on
+EXPOSE 8501
 
+# Run the application
+CMD ["streamlit", "run", "app.py"]