debuging docker

Dockerfile

@@ -1,19 +1,22 @@
-# Use a lightweight PyTorch image with CUDA support
-FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
+# Use a lightweight Python image
+FROM python:3.9
 
 # Set working directory
 WORKDIR /code
 
-# Install dependencies
+# Install system dependencies required for PostgreSQL and transformers
+RUN apt-get update && apt-get install -y \
+    libpq-dev \
+    python3-dev \
+    gcc \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements.txt and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Reduce memory usage (optional)
-ENV TRANSFORMERS_NO_ADVISORY_WARNINGS=1
-ENV HF_HOME="/code/hf_cache"
-
-# Expose API port
+# Expose FastAPI port
 EXPOSE 7860
 
-# Run FastAPI
+# Start FastAPI
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -30,7 +30,7 @@ sentence_transformers
 pinecone-client
 python-multipart
 sqlalchemy
-psycopg2
+psycopg2-binary
 python-dotenv
 torch
 transformers