Update Dockerfile

Dockerfile

@@ -1,32 +1,35 @@
-# Use a specific version of Python (e.g., Python 3.9 for better compatibility with PyTorch)
+# Use an official Python runtime as a parent image
 FROM python:3.9-slim
 
-# Set environment variables to avoid interactive prompts
-ENV DEBIAN_FRONTEND=noninteractive
+# Set the working directory in the container
+WORKDIR /app
 
-RUN apt-get update && apt-get install -y \
-    libgl1-mesa-glx \        # For OpenCV compatibility
-    libglib2.0-0 \           # For OpenCV compatibility
-    build-essential \        # For building certain Python packages \
+# Install system dependencies required for OpenCV
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
+# Copy the requirements.txt into the container at /app
+COPY requirements.txt /app/
 
-# Set the working directory
-WORKDIR /app
-
-# Copy requirements file
-COPY requirements.txt .
+# Install any needed Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Install Python dependencies
-RUN python -m pip install --upgrade pip && \
-    python -m pip install torch torchvision -f https://download.pytorch.org/whl/cpu/torch_stable.html && \
-    python -m pip install -r requirements.txt
+# Create writable directories for caching and configuration
+RUN mkdir -p /app/cache /app/config && chmod -R 777 /app/cache /app/config
 
-# Copy the application code
-COPY . .
+# Set environment variables
+ENV HF_HOME=/app/cache \
+    MPLCONFIGDIR=/app/config/matplotlib \
+    YOLO_CONFIG_DIR=/app/config/ultralytics
+    
+ENV HUGGINGFACE_HUB_ENABLE_HF_UPGRADE_CHECK=false
+# Copy the current directory contents into the container at /app
+COPY . /app/
 
-# Expose the application port
+# Expose port 8000 for the FastAPI app
 EXPOSE 8000
 
-# Command to run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+# Run FastAPI app using uvicorn
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]