Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-# Use Python base image
+# Base image for Python
 FROM python:3.10-slim
 
 # Install system dependencies
@@ -8,25 +8,33 @@ RUN apt-get update && apt-get install -y \
     libsm6 \
     libxext6 \
     cmake \
+    gcc \
+    g++ \
+    ninja-build \
     libgl1-mesa-glx \
     && rm -rf /var/lib/apt/lists/*
 
-# Install pip and essential Python libraries
-RUN pip install --no-cache-dir \
-    torch==1.13.1+cpu \
-    torchvision==0.14.1+cpu \
-    git+https://github.com/facebookresearch/detectron2.git \
-    gradio \
-    requests \
-    opencv-python \
-    Pillow \
-    numpy
+# Upgrade pip
+RUN pip install --no-cache-dir --upgrade pip
+
+# Install PyTorch and torchvision
+RUN pip install --no-cache-dir torch==1.13.1+cpu torchvision==0.14.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
+
+# Install Detectron2
+RUN python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
+
+# Install other Python dependencies
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
 # Set the working directory
 WORKDIR /app
 
-# Copy the application code
+# Copy application files
 COPY . .
 
-# Set the entry point for the container
+# Expose port for Gradio
+EXPOSE 7860
+
+# Run the application
 CMD ["python", "app.py"]