update

Dockerfile

@@ -15,23 +15,26 @@ RUN apt-get update && apt-get install -y \
 # Set working directory
 WORKDIR /app
 
-# Install PyTorch first (CPU version unless you have GPU support in the base image)
-RUN pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2
+# Copy only requirements first to leverage Docker caching
+COPY requirements.txt ./
 
-# Then install mmengine, mmcv, openmim
-RUN pip install mmengine mmcv-full openmim
+# Install PyTorch (CPU version)
+RUN pip install --no-cache-dir torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2
 
-# Then install mmsegmentation
+# Install OpenMMLab dependencies
+RUN pip install --no-cache-dir mmengine mmcv-full openmim
+
+# Install MMSegmentation using mim AFTER torch is present
 RUN mim install mmsegmentation
 
-# Copy your files
+# Copy rest of the application code
 COPY . .
 
-# Install any other Python dependencies
-RUN pip install -r requirements.txt
+# Install remaining Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Expose Gradio port
 EXPOSE 7860
 
-# Run app
+# Start the app
 CMD ["python", "app.py"]