cmarkea
/

bloomz-560m-guardrail

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Cyrile commited on Dec 1, 2023

Commit

41a6ba8

•

1 Parent(s): c159428

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ from transformers import pipeline
 guardrail = pipeline("text-classification", "cmarkea/bloomz-560m-guardrail")
 list_text = [...]
-model(
     list_text,
     return_all_scores=True, # Crucial for assessing all modalities of toxicity!
     function_to_apply='sigmoid' # To ensure obtaining a score between 0 and 1!

 guardrail = pipeline("text-classification", "cmarkea/bloomz-560m-guardrail")
 list_text = [...]
+result = guardrail(
     list_text,
     return_all_scores=True, # Crucial for assessing all modalities of toxicity!
     function_to_apply='sigmoid' # To ensure obtaining a score between 0 and 1!