# Use the official Hugging Face base image
FROM python:latest

# Install system-level dependencies
RUN apt-get update && apt-get install -y \
    libgl1-mesa-glx \
    libglib2.0-0

# Copy requirements.txt into the image
COPY requirements.txt /tmp/requirements.txt

# Install Python dependencies
RUN pip install --no-cache-dir -r /tmp/requirements.txt

# Install PyTorch 1.8.1 and its dependencies from PyTorch's wheel
RUN pip install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

RUN python app.py

# Expose port 7860 to run a Gradio app, for example
EXPOSE 7860