def run_watermark_laion()

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

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

def run_watermark_laion(image):
    if not isinstance(image, list):
        image = [image]
    pixel_values = torch.stack([TRANSFORMS(_image) for _image in image])
    return nn.functional.softmax(MODEL(pixel_values), dim=1)[:, 0]