Update Dockerfile

Dockerfile

@@ -19,16 +19,16 @@ RUN apt-get update && apt-get install -y \
 RUN pip install --upgrade pip
 
 # Install PyTorch (CPU version)
-RUN pip install torch==2.1.0+cpu torchvision==0.16.0+cpu --index-url https://download.pytorch.org/whl/cpu
+RUN pip install torch==2.0.1+cpu torchvision==0.15.2+cpu --index-url https://download.pytorch.org/whl/cpu
 
 # Install Python dependencies
 RUN pip install -r requirements.txt
 
-# Install Detectron2 from GitHub
-RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
+# Install Detectron2 from prebuilt CPU wheel (faster than GitHub build)
+RUN pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch2.0/index.html
 
-# Expose FastAPI port
-EXPOSE 7960
+# Expose Hugging Face default port
+EXPOSE 7860
 
 # Start the app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7960"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]