nicholasKluge
/

TeenyTinyLlama-160m-HateBR

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Dec 25, 2023

Commit

d345da8

•

1 Parent(s): 5ff2a46

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -61,8 +61,8 @@ dataset = DatasetDict({
 model = AutoModelForSequenceClassification.from_pretrained(
     "nicholasKluge/TeenyTinyLlama-162m",
     num_labels=2,
-    id2label={0: "NEGATIVE", 1: "POSITIVE"},
-    label2id={"NEGATIVE": 0, "POSITIVE": 1}
 )
 tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/TeenyTinyLlama-162m")

 model = AutoModelForSequenceClassification.from_pretrained(
     "nicholasKluge/TeenyTinyLlama-162m",
     num_labels=2,
+    id2label={0: "NONTOXIC", 1: "TOXIC"},
+    label2id={"NONTOXIC": 0, "TOXIC": 1}
 )
 tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/TeenyTinyLlama-162m")