updating docker

Dockerfile

@@ -1,4 +1,3 @@
-# Use Python 3.9 image
 FROM python:3.9-slim
 
 # Install system dependencies
@@ -12,21 +11,21 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
-# Install PyTorch (CPU version)
-RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+# Install PyTorch CPU version
+RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu torchaudio==2.0.2+cpu --index-url https://download.pytorch.org/whl/cpu
 
-# Install other Python dependencies
+# Install required Python packages
 RUN pip install opencv-python Pillow gdown gradio pycocotools scikit-image shapely
 
-# Install detectron2 from GitHub
+# Install detectron2 manually
 RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
 
-# Copy your project files
+# Set Matplotlib temp folder
+ENV MPLCONFIGDIR=/tmp/matplotlib
+
+# Copy code
 COPY . /app
 WORKDIR /app
 
-# Set environment variable to avoid matplotlib error
-ENV MPLCONFIGDIR=/tmp/matplotlib
-
 # Run Gradio app
 CMD ["python3", "app.py"]