@ -1,9 +1,12 @@
import pytest
from PIL import Image
from imaginairy.safety import create_safety_score
from imaginairy.utils import get_device
from tests import TESTS_FOLDER
@pytest.mark.skipif(get_device() == "cpu", reason="Too slow to run on CPU")
def test_is_nsfw():
img = Image.open(f"{TESTS_FOLDER}/data/safety.jpg")