def run_nsfw()

in video_processing/modules/nsfw.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def run_nsfw(image):
    if not isinstance(image, list):
        image = [image]
    inputs = PROCESSOR(images=image, return_tensors="pt").to(MODEL.device)
    outputs = MODEL(**inputs).logits
    predicted_labels = outputs.argmax(-1)
    return [MODEL.config.id2label[p.cpu().item()] for p in predicted_labels]