Update Dockerfile

Dockerfile

@@ -1,26 +1,32 @@
-# For more information, please refer to https://aka.ms/vscode-docker-python
-FROM python:3-slim
+# Use a specific version of Python (e.g., Python 3.9 for better compatibility with PyTorch)
+FROM python:3.9-slim
 
-EXPOSE 5002
+# Set environment variables to avoid interactive prompts
+ENV DEBIAN_FRONTEND=noninteractive
 
-# Keeps Python from generating .pyc files in the container
-ENV PYTHONDONTWRITEBYTECODE=1
+# Install required system dependencies
+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
+    && rm -rf /var/lib/apt/lists/*
 
-# Turns off buffering for easier container logging
-ENV PYTHONUNBUFFERED=1
+# Set the working directory
+WORKDIR /app
 
-# Install pip requirements
+# Copy requirements file
 COPY requirements.txt .
-RUN python -m pip install -r requirements.txt
 
-WORKDIR /app
-COPY . /app
+# 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
 
-# Creates a non-root user with an explicit UID and adds permission to access the /app folder
-# For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers
-RUN adduser -u 5678 --disabled-password --gecos "" appuser && chown -R appuser /app
-USER appuser
+# Copy the application code
+COPY . .
 
+# Expose the application port
+EXPOSE 8000
 
-# During debugging, this entry point will be overridden. For more information, please refer to https://aka.ms/vscode-docker-python-debug
-CMD ["gunicorn", "--bind", "0.0.0.0:5002", "weights.convert_safetensor_to_pt:app"]
+# Command to run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]