requirements.txt

# requirements.txt

fastapi
uvicorn[standard]
transformers
sentencepiece

# Install the CPU-only version of PyTorch for smaller size and faster install
# If you have a powerful NVIDIA GPU, you can remove this line and install the CUDA version
torch --index-url https://download.pytorch.org/whl/cpu