Updated

Dockerfile

@@ -1,8 +1,8 @@
 # ==============================================================
-# Tech Disciples AI Backend — Dockerfile
+# Tech Disciples AI Backend — Optimized Dockerfile
 # ==============================================================
 
-# Use lightweight official Python image
+# Use official slim Python 3.10 image
 FROM python:3.10-slim
 
 # Set environment variables
@@ -10,25 +10,33 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     APP_HOME=/app
 
-# Set work directory
+# Set working directory
 WORKDIR $APP_HOME
 
-# System dependencies
-RUN apt-get update && apt-get install -y \
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
+    curl \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirement file first (for caching)
+# Upgrade pip
+RUN pip install --upgrade pip setuptools wheel
+
+# Copy requirements file first (for caching)
 COPY requirements.txt .
 
-# Install Python dependencies
+# Install heavy packages separately first (torch) for reliability
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
+# Install remaining Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application files
+# Copy application code
 COPY . .
 
 # Expose port for FastAPI
 EXPOSE 7860
 
-# Run the FastAPI app with Uvicorn
+# Start FastAPI with Uvicorn
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -1,6 +1,5 @@
 fastapi>=0.101.0
 uvicorn[standard]>=0.23.0
-torch>=2.1.0
 transformers>=4.40.0
 accelerate>=0.25.0
 langchain>=1.0.0