Dockerfile

FROM python:3.9

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

#RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
# Add to your Dockerfile before installing other packages
RUN pip install --no-cache-dir \
    torch==2.0.1 \
    transformers==4.35.2 \
    fastapi==0.104.1 \
    python-multipart==0.0.6

COPY . .

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