pyproject.toml

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "nemotron-graphic-elements-v1"
version = "1.0.0"
description = "NVIDIA Nemotron Graphic Elements v1 - A specialized object detection system for identifying and extracting key elements from charts and graphs"
readme = "README.md"
requires-python = ">=3.8"
license = {text = "NVIDIA Open Model License"}
authors = [
    {name = "NVIDIA Corporation", email = "tviel@nvidia.com"}
]
keywords = ["object-detection", "yolox", "charts", "graphics", "pdf-ingestion", "computer-vision"]
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Image Recognition",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.8",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
]

dependencies = [
    "torch>=2.0.0",
    "numpy>=1.21.0",
    "matplotlib>=3.5.0",
    "pandas>=1.3.0",
    "Pillow>=9.0.0",
]

[project.optional-dependencies]
dev = [
    "jupyter>=1.0.0",
    "notebook>=6.4.0",
]

[project.urls]
Homepage = "https://huggingface.co/nvidia/nemotron-graphic-elements-v1"
Documentation = "https://huggingface.co/nvidia/nemotron-graphic-elements-v1"
Repository = "https://huggingface.co/nvidia/nemotron-graphic-elements-v1"
"Bug Reports" = "https://app.intigriti.com/programs/nvidia/nvidiavdp/detail"

[tool.setuptools]
packages = ["nemotron_graphic_elements_v1", "nemotron_graphic_elements_v1.yolox", "nemotron_graphic_elements_v1.post_processing"]

[tool.setuptools.package-data]
"nemotron_graphic_elements_v1" = ["*.pth", "*.json", "*.png"]