FIX: Unused variables

This commit is contained in:
Евгений Гурьев | Eugene Gourieff | 古仁
2025-01-18 12:57:30 +07:00
parent d6f9250772
commit c20f3a3aff
-3
View File
@@ -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)