Dockerfile

# Use an official Python runtime as a parent image
FROM python:3.8

# Set the working directory in the container to /app
WORKDIR /app

# Copy the current directory contents into the container at /app
ADD . /app
RUN sudo apt-get update -y
RUN sudo apt-get upgrade -y
# Install any needed packages specified in requirements.txt
RUN python -m pip install --upgrade pip
RUN pip install transformers[torch]
RUN pip install -r requirements.txt

RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# Make port 80 available to the world outside this container
#EXPOSE 80

# Run app.py when the container launches
#CMD ["python", "app.py"]

COPY . .

CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]