dvc.yaml

stages:
  model_ingestion:
    cmd: python -c "from components.model_ingestion import ModelIngestor; ModelIngestor.download_model()"
    deps:
      - components/model_ingestion.py
      - utils/commons.py
    outs:
      - models/
      
  serve_model:
    cmd: python main.py
    deps:
      - main.py
      - components/
      - utils/
    outs:
      - logs/