FROM python:3.12-slim

WORKDIR /app

# Install dependencies
RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \
    pip install --no-cache-dir fastapi uvicorn[standard]

# Copy application code
COPY model.py server.py ./
COPY model_weights/ model_weights/

EXPOSE 7860

CMD ["python", "server.py"]