Metadata-Version: 2.4
Name: cosmos-guardrail
Version: 0.3.1
Summary: Guardrail system for content safety in text and video, part of NVIDIA Cosmos project
Project-URL: Source, https://github.com/nvidia-cosmos/cosmos-predict1
Project-URL: Research, https://research.nvidia.com/labs/dir/cosmos-predict1
Author-email: NVIDIA <cosmos-license@nvidia.com>
License: Apache-2.0
License-File: LICENSE
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10
Requires-Dist: attrs>=25.1.0
Requires-Dist: better-profanity>=0.7.0
Requires-Dist: huggingface-hub>=0.29.2
Requires-Dist: imageio-ffmpeg>=0.4.5
Requires-Dist: imageio>=2.37.0
Requires-Dist: nltk>=3.9.1
Requires-Dist: numpy>=1.26.4
Requires-Dist: opencv-python>=4.10.0.84
Requires-Dist: peft>=0.14.0
Requires-Dist: pillow>=11.1.0
Requires-Dist: protobuf
Requires-Dist: requests>=2.28.0
Requires-Dist: retinaface-py>=0.0.2
Requires-Dist: safetensors>=0.5.3
Requires-Dist: scikit-image>=0.25.2
Requires-Dist: sentencepiece>=0.2.0
Requires-Dist: torch>=2.6.0
Requires-Dist: torchvision>=0.21.0
Requires-Dist: tqdm>=4.66.5
Requires-Dist: transformers>=5.0.0
Description-Content-Type: text/markdown

# cosmos-guardrail

A PyPI-installable package of NVIDIA's Cosmos Content Safety Guardrails  
(Checking text and blurring faces in video for unsafe content).

the code is copy & pasted from NVIDIA's Cosmos projects:

- https://github.com/nvidia-cosmos/cosmos-transfer1  

All credit & copyright belong to NVIDIA

See [Example](./example_use.py) for usage.

## Install

Install the package with uv:

```shell
just install
```

## Test

```shell
just test
```

## Publish

```shell
just publish
```
