Update Dockerfile

Dockerfile

@@ -18,32 +18,30 @@ ENV OMP_NUM_THREADS=1 \
     NVIDIA_VISIBLE_DEVICES=all \
     NVIDIA_DRIVER_CAPABILITIES=compute,utility
 
+# Base OS deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3 python3-pip git git-lfs ca-certificates && \
     rm -rf /var/lib/apt/lists/* && \
     git lfs install
 
+# Modern torch stack aligned to cu121 and Transformers 4.44.x
 RUN python3 -m pip install --upgrade pip
-
 RUN python3 -m pip install --no-cache-dir \
     --index-url https://download.pytorch.org/whl/cu121 \
-    torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
-
-RUN python3 -m pip install --no-cache-dir faiss-gpu==1.7.2
+    torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1
 
-RUN python3 -m pip install --no-cache-dir \
-    sentence-transformers==2.2.2 \
-    huggingface_hub>=0.14.1
+# NOTE: Do NOT install faiss-gpu. We keep CPU FAISS only (from requirements.txt).
+# (Removing: python3 -m pip install faiss-gpu==1.7.2)
 
+# App requirements (single source of truth)
 COPY requirements.txt .
 RUN python3 -m pip install --no-cache-dir -r requirements.txt
 
-RUN python3 -m pip install --no-cache-dir --upgrade --upgrade-strategy eager peft==0.12.0
-
+# App code
 COPY . .
 
+# Needed dirs
 RUN mkdir -p "$OFFLOAD_DIR" "$MERGED_MODEL_DIR" "$HF_HOME" "$TRANSFORMERS_CACHE"
 
 EXPOSE 7860
-
-CMD ["python3", "app.py"]
+CMD ["python3", "app.py"]