def load_nsfw()

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

• 4 lines of code
• 1 McCabe index (conditional complexity)

def load_nsfw(device):
    global MODEL, PROCESSOR
    MODEL = AutoModelForImageClassification.from_pretrained(MODEL_ID).eval().to(device)
    PROCESSOR = AutoImageProcessor.from_pretrained(MODEL_ID)