Docker and req.txt changes

Dockerfile

@@ -45,16 +45,19 @@ ENV HOME=/home/user \
 
 WORKDIR /app
 
-# Install PyTorch 1.9 CPU
+# Install PyTorch 1.9 CPU first
 RUN pip install --user torch==1.9.0+cpu torchvision==0.10.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
 # Install numpy first (required for other packages)
 RUN pip install --user numpy==1.21.6
 
-# Install core dependencies
+# Install core dependencies in order
 RUN pip install --user \
     Pillow==8.3.2 \
-    opencv-python==4.5.5.64 \
+    opencv-python==4.5.5.64
+
+# Install scientific computing dependencies
+RUN pip install --user \
     scipy==1.7.3 \
     scikit-image==0.19.3 \
     scikit-learn==1.0.2
@@ -62,10 +65,10 @@ RUN pip install --user \
 # Install detectron2 for PyTorch 1.9 CPU
 RUN pip install --user detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.9/index.html
 
-# Install pycocotools with pre-built wheel to avoid compilation issues
+# Install pycocotools separately to avoid compilation issues
 RUN pip install --user pycocotools --no-build-isolation
 
-# Install other ML dependencies
+# Install ML dependencies
 RUN pip install --user \
     timm==0.4.12 \
     einops==0.6.1 \
@@ -74,13 +77,20 @@ RUN pip install --user \
     tqdm==4.64.1 \
     imutils==0.5.4
 
-# Install Gradio and web dependencies with compatible versions
+# Install web framework dependencies with compatible versions
 RUN pip install --user \
-    gradio==3.35.2 \
-    huggingface_hub==0.11.1 \
+    httpx==0.23.0 \
+    httpcore==0.15.0 \
+    anyio==3.6.1 \
+    starlette==0.19.1 \
     fastapi==0.78.0 \
     uvicorn==0.18.2
 
+# Install Gradio and HuggingFace Hub with compatible versions
+RUN pip install --user \
+    huggingface_hub==0.8.1 \
+    gradio==3.1.7
+
 # Install remaining dependencies
 RUN pip install --user \
     PyYAML==5.4.1 \
@@ -91,10 +101,6 @@ RUN pip install --user \
     gdown==4.5.4 \
     wget==3.2
 
-# Try NATTEN for CPU (optional - will continue if fails)
-RUN pip install --user natten==0.14.6 -f https://shi-labs.com/natten/wheels/cpu/torch1.9/index.html || \
-    echo "NATTEN installation failed - continuing without it"
-
 # Copy application files
 COPY --chown=user:user . /app
 

requirements.txt

@@ -2,7 +2,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch==1.9.0+cpu
 torchvision==0.10.0+cpu
-torchaudio==0.9.0
 
 # Core dependencies
 numpy==1.21.6
@@ -15,32 +14,21 @@ detectron2==0.6
 # OpenCV
 opencv-python==4.5.5.64
 
-# Git dependencies
-
 # Image utilities
 imutils==0.5.4
 
-# NATTEN CPU version for torch 1.9
--f https://shi-labs.com/natten/wheels/cpu/torch1.9/index.html
-natten==0.14.6
-
-# Build tools
-cython==0.29.35
-setuptools==59.5.0
-
 # Scientific computing
 scipy==1.7.3
 shapely==1.8.5
 h5py==3.7.0
 scikit-image==0.19.3
+scikit-learn==1.0.2
 
 # Deep learning utilities
 timm==0.4.12
 einops==0.6.1
 
 # Development tools
-icecream==2.1.3
-wandb==0.13.5
 tqdm==4.64.1
 
 # NLP utilities
@@ -51,18 +39,28 @@ inflect==6.0.4
 # File utilities
 gdown==4.5.4
 wget==3.2
-huggingface_hub==0.11.1
 
-# Training framework
-pytorch_lightning==1.8.6
+# Compatible web interface versions
+gradio==3.1.7
+huggingface_hub==0.8.1
+fastapi==0.78.0
+uvicorn==0.18.2
+
+# YAML support
+PyYAML==5.4.1
+
+# Matplotlib
+matplotlib==3.5.3
+
+# Build tools
+cython==0.29.35
+setuptools==59.5.0
+
+# COCO API
+pycocotools
 
-# Web interface
-gradio==3.35.2
-scikit-learn
-diffdist
-gdown
-tqdm
-huggingface_hub
-wget
-# Job submission
-submitit==1.4.5
+# Additional gradio dependencies
+httpx==0.23.0
+httpcore==0.15.0
+anyio==3.6.1
+starlette==0.19.1