Update Dockerfile and requirements.txt to resolve torch/vllm dependency conflict

Dockerfile

@@ -1,20 +1,25 @@
-FROM python:3.13.5-slim
+FROM python:3.12-slim
 
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
     build-essential \
-    curl \
-    git \
+    libpq-dev \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+COPY requirements.txt .
 
-RUN pip3 install -r requirements.txt
+# Install vLLM with GPU support first (fetches compatible Torch)
+RUN pip install --no-cache-dir vllm==0.10.1 --extra-index-url https://download.pytorch.org/whl/cu128
 
-EXPOSE 8501
+# Install remaining dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py pipeline.py db_utils.py ./
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+ENV HF_HOME=/cache/huggingface
+ENV PORT=8501
+
+EXPOSE 8501
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]

requirements.txt

@@ -1,7 +1,6 @@
-torch==2.8.0
 transformers==4.56.0
 accelerate==1.10.1
 psycopg2-binary==2.9.10
 sqlalchemy==2.0.43
 python-dotenv==1.1.1
-vllm==0.6.2
+vllm==0.10.1