Update Dockerfile

Dockerfile

@@ -1,24 +1,24 @@
-# Step 1: Use an appropriate base image
+# Use an appropriate base image (you can adjust this as per your requirements)
 FROM python:3.10
 
-# Step 2: Switch to root user
+# Switch to root user
 USER root
 
-# Step 3: Install necessary dependencies
+# Install system dependencies
 RUN chmod 1777 /tmp \
     && apt update -q && apt install -y ca-certificates wget libgl1 \
     && wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
     && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
     && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
 
-# Step 4: Install FastAPI and other Python dependencies
-RUN pip install fastapi[all] uvicorn
+# Install Python dependencies
+RUN pip install fastapi[all] uvicorn Pillow
 
-# Step 5: Copy application code
+# Copy the application code
 COPY main.py main.py
 
-# Step 6: Run Python script to check for errors before starting FastAPI
+# Run Python script to check for errors before starting FastAPI
 RUN python main.py
 
-# Step 7: Start FastAPI server
+# Start FastAPI server
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]