cog.yaml

build:
  gpu: false
  python_version: "3.10"
  python_packages:
    - "openai>=1.0.0"
    - "fastapi>=0.110.0"
    - "uvicorn>=0.27.0"
    - "pydantic>=2.5.0"
    - "python-dotenv>=1.0.0"
    - "faiss-cpu>=1.7.4"
    - "numpy>=1.24.0"
    - "scikit-learn>=1.3.0"
    - "pandas>=2.0.0"
    - "torch>=2.0.0"
    - "langchain>=0.0.335"

# The prediction interface
predict: "replicate.py:Predictor"

# Data files needed for inference
# These will be included in the Docker image
image:
  packages:
    - "curl"
    - "wget"
    - "git"

# Specify how much memory is needed
resources:
  cpu: 2
  memory: 8GB