FIX: Unused variables
This commit is contained in:
parent
d6f9250772
commit
c20f3a3aff
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user