Update Dockerfile

Dockerfile

@@ -1,13 +1,30 @@
-FROM python:3.9
+# Use a slim Python base image
+FROM python:3.9-slim
 
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
+# Install system dependencies for OpenCV
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    git
 
+# Set the working directory
 WORKDIR /app
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+# Copy the requirements file and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-COPY --chown=user . /app
-CMD ["ginicorn", "main:app", "-b", "0.0.0.0:7860"]
+# Copy your application code
+COPY . .
+
+# Download the YOLOv8 model (alternatively, ensure it's copied with your app)
+RUN python -c "from ultralytics import YOLO; YOLO('yolov8n.pt')"
+
+# Expose the port Hugging Face Spaces expects
+EXPOSE 7860
+
+# Command to run your Flask app with Gunicorn
+CMD ["gunicorn", "-w", "4", "-b", "0.0.0.0:7860", "app:app"]