gaidasalsaa's picture
Fix: Locked Transformers version to 4.35.2 for PyTorch 2.1 compatibility
806ab2d
raw
history blame contribute delete
307 Bytes
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
WORKDIR /app
COPY requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir fastapi uvicorn[standard] pydantic "transformers==4.35.2" tweepy numpy
COPY . /app
ENV PORT 7860
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]