diff --git a/scripts/reactor_sfw.py b/scripts/reactor_sfw.py index ec1d977..029ea55 100644 --- a/scripts/reactor_sfw.py +++ b/scripts/reactor_sfw.py @@ -6,9 +6,6 @@ SCORE = 0.85 logging.getLogger('transformers').setLevel(logging.ERROR) -img_path = "test.jpg" -model_path = "models/vit-base-nsfw-detector" - def nsfw_image(img_path: str, model_path: str): img = Image.open(img_path) predict = pipeline("image-classification", model=model_path)