# Use the official Python image as a base
FROM python:3.9

# Set the working directory
WORKDIR /app

# Copy the requirements file into the container
COPY requirements.txt .

# Install the dependencies
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install accelerate

# Copy the rest of the application code into the container
COPY . .

# Expose the port FastAPI will run on
EXPOSE 7860

# Command to run the application
CMD ["sudo", "python", "app.py"]