from NoInsult import predict result = predict("I will fuck you up, you piece of shit.") print(result["flags"]) # {'toxicity': True, 'severe_toxicity': False, ...} print(result["scores"]) # raw sigmoid probabilities per label