Update Dockerfile

Dockerfile

@@ -9,6 +9,15 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
+# Copy requirements first (for better Docker layer caching)
+COPY requirements.txt .
+
+# Install PyTorch first to avoid conflicts
+RUN pip install --no-cache-dir torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cpu
+
+RUN pip install pip-tools
+RUN pip-compile requirements.txt
+RUN pip install -r requirements.txt
 
 # Clone and install MobileCLIP
 RUN git clone https://github.com/apple/ml-mobileclip.git && \
@@ -17,17 +26,6 @@ RUN git clone https://github.com/apple/ml-mobileclip.git && \
     cd .. && \
     rm -rf ml-mobileclip/.git
 
-# Install packages from git (if you need latest versions)
-RUN pip install git+https://github.com/mlfoundations/open_clip.git
-RUN pip install git+https://github.com/huggingface/pytorch-image-models.git
-
-# Copy requirements first (for better Docker layer caching)
-COPY requirements.txt .
-
-# Install Python dependencies (excluding packages we'll install from git)
-RUN pip install --no-cache-dir -r requirements.txt
-
-
 # Copy application code
 COPY app.py .