Sk1306
/

student_chat_toxicity_classifier_model

Text Classification

Model card Files Files and versions

Sk1306 commited on Jan 25

Commit

67eaea7

·

verified ·

1 Parent(s): ed86227

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -55,7 +55,7 @@ import torch
 from transformers import RobertaTokenizer, RobertaForSequenceClassification
 # Load the model and tokenizer
-model_name = "path/to/your/model/directory"
 tokenizer = RobertaTokenizer.from_pretrained(model_name)
 model = RobertaForSequenceClassification.from_pretrained(model_name)

 from transformers import RobertaTokenizer, RobertaForSequenceClassification
 # Load the model and tokenizer
+model_name = "Sk1306/student_chat_toxicity_classifier_model"
 tokenizer = RobertaTokenizer.from_pretrained(model_name)
 model = RobertaForSequenceClassification.from_pretrained(model_name)