FROM python:3.10-slim

WORKDIR /app

# Install specific versions
RUN pip install --no-cache-dir \
    torch==2.5.1 \
    transformers==4.57.1 \
    accelerate \
    gradio>=4.0.0 \
    huggingface_hub \
    bitsandbytes \
    pillow

COPY app.py .

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