OWASP-AIBOM-Generator / PROJECT_README.md
e2hln's picture
Upload 44 files
6165ba9 verified
|
raw
history blame
3.37 kB

πŸ€– OWASP GenAI Security Project - AIBOM Generator

This is the official GitHub repository for the OWASP AIBOM Generator β€” an open-source tool for generating AI Bills of Materials (AIBOMs) in CycloneDX format.
The tool is also listed in the official CycloneDX Tool Center.

πŸš€ Try the tool live:
πŸ‘‰ https://owasp-genai-aibom.org
πŸ”– Bookmark and share: https://owasp-genai-aibom.org

🌐 OWASP AIBOM Initiative: genai.owasp.org

This initiative is about making AI transparency practical. The OWASP AIBOM Generator, running under the OWASP GenAI Security Project, is focused on helping organizations understand what’s actually inside AI models and systems, starting with open models on Hugging Face. Join OWASP GenAI Security Project - AIBOM Initiative to contribute.


πŸ“¦ What It Does

  • Extracts metadata from models hosted on Hugging Face πŸ€—
  • Generates an AIBOM (AI Bill of Materials) in CycloneDX 1.6 JSON format
  • Calculates AIBOM completeness scoring with recommendations
  • Supports metadata extraction from model cards, configurations, and repository files

πŸ›  Features

  • Human-readable AIBOM viewer
  • JSON download
  • Completeness scoring & improvement tips
  • API endpoints for automation
  • Standards-aligned generation (CycloneDX 1.6, compatible with SPDX AI Profile)

οΏ½ Installation & Usage

1. Install Dependencies

pip install -r requirements.txt

Or, if you prefer uv for faster dependency management:

uv sync

2. Run Web Application

Start the local server at http://localhost:8000:

python3 -m src.main

3. Run via CLI

Generate an AIBOM for a Hugging Face model directly from your terminal:

Basic Usage:

python3 -m src.cli google-bert/bert-base-uncased

Advanced Usage: You can specify additional metadata like component name, version, and supplier.

python3 -m src.cli google-bert/bert-base-uncased \
  --name "My Custom BERT" \
  --version "1.0.0" \
  --manufacturer "Acme Corp" \
  --output "my_sbom.json"

Command Line Options:

Option Shorthand Description
model_id Hugging Face Model ID (e.g., owner/model)
--test -t Run test mode for multiple predefined models
--output -o Custom output file path
--name -n Override component name in metadata
--version -v Override component version in metadata
--manufacturer -m Override component manufacturer/supplier
--inference -i Use AI inference for enhanced metadata (requires API key)
--summarize -s Enable intelligent description summarization
--verbose Enable verbose logging
  • Metrics and produced SBOMs are saved to the sboms/ directory by default.

�🐞 Found a Bug or Have an Improvement Request?

We welcome contributions and feedback.

➑ Log an issue:
https://github.com/GenAI-Security-Project/aibom-generator/issues


πŸ“„ License

This project is open-source and available under the Apache 2.0 License.