abdoolamunir's picture
Update hugginface.yml
ecc5f2a verified
# huggingface.yml
build:
dockerfile: Dockerfile # Path to Dockerfile from the root of the repository
context: . # Docker build context
runtime:
env: # Environment variables needed by the application
TRANSFORMERS_CACHE: /app/.cache/huggingface/transformers # Cache directory for Transformers
port: 80 # The port on which your FastAPI application is configured to run
resources:
cpu: 2 # Number of CPU cores
memory: 4G # Amount of memory allocated (4 GB in this example)
gpu: 0 # Number of GPUs (0 for CPU only)